Update app.py

app.py

@@ -1,28 +1,21 @@
 import streamlit as st
-import openai
 import google.generativeai as genai
 
 # --------------------------
 # 1. Page Configuration
 # --------------------------
-st.set_page_config(page_title="AI-Powered Code Reviewer", layout="wide")
-st.title("🐍 AI-Powered Python Code Reviewer")
+st.set_page_config(page_title="AI-Powered Code Reviewer (Gemini)", layout="wide")
+st.title("🐍 AI-Powered Python Code Reviewer (Gemini)")
 
 # --------------------------
-# 2. Sidebar API Selection
+# 2. Configure Gemini API from Secrets
 # --------------------------
-st.sidebar.header("⚙️ Settings")
-
-api_choice = st.sidebar.radio("Choose API:", ["OpenAI", "Google Gemini"])
-
-if api_choice == "OpenAI":
-    api_key = st.sidebar.text_input("Enter OpenAI API Key", type="password")
-    if api_key:
-        openai.api_key = api_key
+if "GEMINI_API_KEY" in st.secrets:
+    api_key = st.secrets["GEMINI_API_KEY"]
+    genai.configure(api_key=api_key)
 else:
-    api_key = st.sidebar.text_input("Enter Google API Key", type="password")
-    if api_key:
-        genai.configure(api_key=api_key)
+    st.error("❌ Gemini API key not found! Please add it in Hugging Face Secrets.")
+    st.stop()
 
 # --------------------------
 # 3. User Code Input
@@ -33,21 +26,20 @@ user_code = st.text_area("Paste Python code here:", height=300)
 # --------------------------
 # 4. Code Review Function
 # --------------------------
-def review_code_with_openai(code):
-    response = openai.ChatCompletion.create(
-        model="gpt-4o-mini",   # You can change model (gpt-4o, gpt-4o-mini)
-        messages=[
-            {"role": "system", "content": "You are a Python code reviewer. Identify bugs and suggest fixes."},
-            {"role": "user", "content": f"Review this Python code:\n{code}"}
-        ],
-        temperature=0.3,
-    )
-    return response["choices"][0]["message"]["content"]
-
 def review_code_with_gemini(code):
-    model = genai.GenerativeModel("gemini-2.5-pro")
+    model = genai.GenerativeModel("gemini-1.5-pro")  # ✅ Gemini 1.5 Pro
     response = model.generate_content(
-        f"Review this Python code, detect bugs, and suggest fixes with corrected code:\n{code}"
+        f"""
+        You are an expert Python code reviewer.
+        Analyze the following code for:
+        - Bugs
+        - Logical errors
+        - Improvements in readability and efficiency
+        - Provide corrected code if necessary
+
+        Code:
+        {code}
+        """
     )
     return response.text
 
@@ -57,16 +49,11 @@ def review_code_with_gemini(code):
 if st.button("🔍 Review Code"):
     if not user_code.strip():
         st.warning("⚠️ Please enter some Python code first.")
-    elif not api_key:
-        st.error("❌ Please enter your API key in the sidebar.")
     else:
-        with st.spinner("Analyzing your code..."):
+        with st.spinner("Analyzing your code with Gemini..."):
             try:
-                if api_choice == "OpenAI":
-                    feedback = review_code_with_openai(user_code)
-                else:
-                    feedback = review_code_with_gemini(user_code)
-                
+                feedback = review_code_with_gemini(user_code)
+
                 st.subheader("📢 Review Feedback & Fixes")
                 st.write(feedback)