Update app.py

app.py

@@ -1,61 +1,59 @@
-import streamlit as st
-import google.generativeai as genai
-
-# --------------------------
-# 1. Page Configuration
-# --------------------------
-st.set_page_config(page_title="AI-Powered Code Reviewer (Gemini)", layout="wide")
-st.title("🐍 AI-Powered Python Code Reviewer (Gemini)")
-
-# --------------------------
-# 2. Configure Gemini API from Secrets
-# --------------------------
-if "GEMINI_API_KEY" in st.secrets:
-    api_key = st.secrets["GEMINI_API_KEY"]
-    genai.configure(api_key=api_key)
-else:
-    st.error("❌ Gemini API key not found! Please add it in Hugging Face Secrets.")
-    st.stop()
-
-# --------------------------
-# 3. User Code Input
-# --------------------------
-st.subheader("✍️ Enter Your Python Code")
-user_code = st.text_area("Paste Python code here:", height=300)
-
-# --------------------------
-# 4. Code Review Function
-# --------------------------
-def review_code_with_gemini(code):
-    model = genai.GenerativeModel("gemini-1.5-pro")  # ✅ Gemini 1.5 Pro
-    response = model.generate_content(
-        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
-
-# --------------------------
+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") 
+# -------------------------- 
+# 2. Sidebar API Selection 
+# -------------------------- 
+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 
+    else: 
+        api_key = st.sidebar.text_input("Enter Google API Key", type="password") 
+        if api_key: 
+            genai.configure(api_key=api_key) 
+# -------------------------- 
+# 3. User Code Input 
+# -------------------------- 
+st.subheader("✍️ Enter Your Python Code") 
+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") 
+    response = model.generate_content( f"Review this Python code, detect bugs, and suggest fixes with corrected code:\n{code}" ) 
+    return response.text 
+# -------------------------- 
 # 5. Run Code Review
-# --------------------------
-if st.button("🔍 Review Code"):
-    if not user_code.strip():
-        st.warning("⚠️ Please enter some Python code first.")
-    else:
-        with st.spinner("Analyzing your code with Gemini..."):
-            try:
-                feedback = review_code_with_gemini(user_code)
-
-                st.subheader("📢 Review Feedback & Fixes")
-                st.write(feedback)
-
-            except Exception as e:
-                st.error(f"Error: {str(e)}")
+# -------------------------- 
+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..."): 
+            try: 
+                if api_choice == "OpenAI": 
+                    feedback = review_code_with_openai(user_code) 
+                else: 
+                    feedback = review_code_with_gemini(user_code) 
+                    st.subheader("📢 Review Feedback & Fixes") 
+                    st.write(feedback) 
+            except Exception as e: 
+                st.error(f"Error: {str(e)}")