kumar1907/Code_Review_Agent

1import streamlit as st2from code_reviewer import review_code3from pr_fetcher import fetch_pr_diff4 5st.set_page_config(page_title="Code Review Agent", layout="wide")6st.title("🤖 Code Review Agent (Hugging Face Edition)")7st.caption("Fetch a Pull Request from GitHub and get AI-powered review + improved code")8 9# Sidebar for inputs10st.sidebar.header("🔗 GitHub PR Details")11repo_owner = st.sidebar.text_input("GitHub Repo Owner", placeholder="e.g. openai")12repo_name = st.sidebar.text_input("GitHub Repo Name", placeholder="e.g. gpt-code-review")13pr_number = st.sidebar.number_input("PR Number", min_value=1, step=1)14github_token = st.sidebar.text_input("GitHub Token (optional if public repo)", type="password")15 16if st.sidebar.button("🚀 Fetch & Review PR"):17    with st.spinner("Fetching PR diff..."):18        diff_text = fetch_pr_diff(repo_owner, repo_name, pr_number, github_token)19 20    if diff_text.startswith("Error"):21        st.error(diff_text)22    else:23        st.success("✅ PR diff fetched successfully!")24        25        # Display the diff for reference26        with st.expander("🔍 View Raw PR Diff"):27            st.code(diff_text, language="diff")28 29        # Review with LLM30        with st.spinner("Analyzing code with Hugging Face model..."):31            feedback, improved_code = review_code(diff_text)32 33        # Show results34        st.subheader("💡 Review Feedback")35        st.markdown(feedback)36 37        st.subheader("✨ Suggested Improved Code")38        st.code(improved_code, language="python")39