Update app.py

app.py

@@ -2,36 +2,73 @@ import streamlit as st
 import google.generativeai as genai
 import os
 
-# Configure API
-genai.configure(api_key="AIzaSyBiAW2GQLid0HGe9Vs_ReKwkwsSVNegNzs")
+# ---------- UI ----------
+st.set_page_config(page_title="Python Code Reviewer 🤖", layout="centered")
+st.title("🐍 Python Code Reviewer 🤖")
+st.markdown("### Paste your code and a short prompt – the AI will review it for you!")
 
-# Define system prompt
-sys_prompt = os.getenv("USER_PROMPT")
+# 1️⃣ API key input (optional)
+api_key_input = st.text_input(
+    "🔑 Google API Key (optional – leave blank to use the built‑in key)",
+    type="password",
+    help="If you have your own Gemini API key, enter it here. Otherwise the app will fall back to the default key."
+)
 
-# Initialize the model
-model = genai.GenerativeModel(model_name="models/gemini-2.5-flash", system_instruction=sys_prompt)
+# 2️⃣ System prompt (optional)
+sys_prompt = st.text_input(
+    "🗒️ System Prompt (optional)",
+    placeholder="e.g., 'You are a friendly Python code reviewer.'",
+    help="Custom instructions for the model. If left empty, the environment variable USER_PROMPT is used."
+)
 
-# Streamlit UI Configuration
-st.set_page_config(page_title="Python Code Reviewer 🤖", layout="centered")
+# 3️⃣ Code box
+code_box = st.text_area(
+    "💻 Your Python code:",
+    height=200,
+    placeholder="# Paste your Python code here"
+)
 
-# Title & Introduction
-st.title("🐍 Python Code Reviewer 🤖")
-st.markdown("### Welcome! I can help you debug and improve your Python code. Just paste your code below! 🚀")
-
-# Input Text Area
-user_query = st.text_area("💬 Enter your Python-related code snippet or query:", height=200)
-
-# Generate Response Button
-if st.button("🚀 Review Code"):
-    if user_query.strip():
-        with st.spinner("Analyzing your code... 🧐"):
-            response = model.generate_content(user_query)
-        
-        # Display Response
-        st.markdown("### 🤖 Response:")
-        st.success(response.text)
-    else:
-        st.warning("⚠️ Please enter a Python-related code snippet!")
+# 4️⃣ Action prompt box
+prompt_box = st.text_area(
+    "📝 What should the AI do with this code?",
+    height=80,
+    placeholder="e.g., 'Find bugs and suggest improvements.'"
+)
+
+# ---------- Logic ----------
+if st.button("🚀 Review"):
+    # Validate that there is code to review
+    if not code_box.strip():
+        st.warning("⚠️ Please paste some Python code.")
+        st.stop()
+
+    # Choose API key: user‑provided > hard‑coded fallback
+    api_key = api_key_input.strip() or "AIzaSyBiAW2GQLid0HGe9Vs_ReKwkwsSVNegNzs"
+    genai.configure(api_key=api_key)
+
+    # Choose system instruction: user input > env var > none
+    system_instruction = (
+        sys_prompt.strip()
+        or os.getenv("USER_PROMPT")
+        or "You are a helpful Python code reviewer."
+    )
+
+    # Initialise model with the chosen system instruction
+    model = genai.GenerativeModel(
+        model_name="models/gemini-2.5-flash",
+        system_instruction=system_instruction,
+    )
+
+    # Build the full user message
+    user_message = f"Code:\n```python\n{code\_box}\n```\n\nTask: {prompt_box or 'Review the code.'}"
+
+    # Call Gemini
+    with st.spinner("Analyzing your code... 🧐"):
+        response = model.generate_content(user_message)
+
+    # Show result
+    st.markdown("### 🤖 Response")
+    st.success(response.text)
 
 # Footer
 st.markdown("---")