Update app.py

app.py

@@ -2,65 +2,37 @@ import streamlit as st
 from code_reviewer import review_code
 from pr_fetcher import fetch_pr_diff
 
-# -----------------------------
-# Page Config (must be first Streamlit command)
-# -----------------------------
-st.set_page_config(
-    page_title="Code Review Agent",
-    page_icon="🤖",
-    layout="wide"
-)
-
-# -----------------------------
-# App Title
-# -----------------------------
+st.set_page_config(page_title="Code Review Agent", layout="wide")
 st.title("🤖 Code Review Agent (Hugging Face Edition)")
-st.write("Fetch a Pull Request from GitHub and get **AI-powered review + improved code**")
+st.caption("Fetch a Pull Request from GitHub and get AI-powered review + improved code")
 
-# -----------------------------
-# Sidebar Inputs
-# -----------------------------
-with st.sidebar:
-    st.header("📂 GitHub PR Details")
-    owner = st.text_input("GitHub Repo Owner", "kumar1905")
-    repo = st.text_input("GitHub Repo Name", "TEMPERATURE-CONVERTOR")
-    pr_number = st.number_input("PR Number", min_value=1, value=1, step=1)
-    token = st.text_input("GitHub Token (optional if public repo)", type="password")
-    fetch_btn = st.button("🚀 Fetch & Review PR")
+# Sidebar for inputs
+st.sidebar.header("🔗 GitHub PR Details")
+repo_owner = st.sidebar.text_input("GitHub Repo Owner", placeholder="e.g. openai")
+repo_name = st.sidebar.text_input("GitHub Repo Name", placeholder="e.g. gpt-code-review")
+pr_number = st.sidebar.number_input("PR Number", min_value=1, step=1)
+github_token = st.sidebar.text_input("GitHub Token (optional if public repo)", type="password")
 
-# -----------------------------
-# Fetch + Review
-# -----------------------------
-if fetch_btn:
+if st.sidebar.button("🚀 Fetch & Review PR"):
     with st.spinner("Fetching PR diff..."):
-        diff = fetch_pr_diff(owner, repo, pr_number, token)
-
-        if diff.startswith("Error"):
-            st.error(diff)
-        else:
-            st.success("✅ PR diff fetched successfully!")
-            with st.expander("📄 View Raw PR Diff"):
-                st.code(diff, language="diff")
-
-            with st.spinner("Analyzing code with Hugging Face model..."):
-                feedback, code = review_code(diff)
-
-            # Feedback Section
-            st.subheader("💡 Review Feedback")
-            if feedback:
-                st.write(feedback)
-            else:
-                st.warning("No feedback generated.")
-
-            # Improved Code Section
-            st.subheader("✨ Suggested Improved Code")
-            if code:
-                st.code(code, language="python")
-            else:
-                st.warning("No improved code generated.")
-
-# -----------------------------
-# Footer 
-# -----------------------------
-st.markdown("---")
-st.markdown("Built with ❤️ using [Streamlit](https://streamlit.io/) & [Hugging Face Transformers](https://huggingface.co/)")
+        diff_text = fetch_pr_diff(repo_owner, repo_name, pr_number, github_token)
+
+    if diff_text.startswith("Error"):
+        st.error(diff_text)
+    else:
+        st.success("✅ PR diff fetched successfully!")
+        
+        # Display the diff for reference
+        with st.expander("🔍 View Raw PR Diff"):
+            st.code(diff_text, language="diff")
+
+        # Review with LLM
+        with st.spinner("Analyzing code with Hugging Face model..."):
+            feedback, improved_code = review_code(diff_text)
+
+        # Show results
+        st.subheader("💡 Review Feedback")
+        st.markdown(feedback)
+
+        st.subheader("✨ Suggested Improved Code")
+        st.code(improved_code, language="python")