Update app.py

app.py

@@ -6,56 +6,49 @@ from google import genai
 # Initialize Gemini Client with API key from environment
 client = genai.Client(api_key=os.environ.get("GEMINI_API_KEY"))
 
-# Helper to encode image for Gemini
-def process_image(image_file):
-    image_bytes = image_file.read()
-    encoded_image = base64.b64encode(image_bytes).decode("utf-8")
-    return {
-        "inline_data": {
-            "mime_type": image_file.type,
-            "data": encoded_image
-        }
-    }
+# Supported file types
+SUPPORTED_EXTENSIONS = ["py", "c", "cpp", "java", "js", "jsx"]
+
+# Helper: Read and decode code file
+def read_code_file(file):
+    content = file.read().decode("utf-8")
+    filename = file.name
+    return filename, content
 
 # Main App
 def main():
-    st.set_page_config(page_title="🌿 Leaf Disease Detector", layout="centered")
-    st.title("🌱 Leaf Disease Detector")
-    st.write("Upload an image of a plant leaf, and we'll analyze it to detect possible diseases, "
-             "provide treatment suggestions, and find real-world statistics from the internet about this disease.")
-
-    uploaded_image = st.file_uploader("Upload a leaf image (JPG or PNG)", type=["jpg", "jpeg", "png"])
-
-    if uploaded_image:
-        st.image(uploaded_image, caption="Uploaded Leaf", use_container_width=True)
-        image_data = process_image(uploaded_image)
-
-        if st.button("🧪 Detect Disease"):
-            with st.spinner("Analyzing leaf and searching for information..."):
-                content_blocks = [
-                    {
-                        "text": (
-                            "You are a plant disease diagnostic AI. Analyze the uploaded leaf image and do the following:\n\n"
-                            "1. Identify any plant disease visible on the leaf.\n"
-                            "2. Provide the disease name and a short scientific description.\n"
-                            "3. Suggest treatment or prevention methods farmers can use.\n"
-                            "4. Use **Google Search** to find:\n"
-                            "   - Estimated global or regional financial losses due to this disease\n"
-                            "   - Impactful statistics or facts due to this disease (e.g., crops affected, common locations, yield reduction, etc.)\n"
-                            "If the leaf looks healthy, clearly state that."
-                        )
-                    },
-                    image_data
-                ]
+    st.set_page_config(page_title="🧠 Code Analyzer & Fixer", layout="centered")
+    st.title("🛠️ Code Analyzer & Bug Fixer")
+    st.write("Upload a code file (Python, C, C++, Java, or React) and get a detailed analysis including:\n"
+             "- README-style description\n"
+             "- Bug detection\n"
+             "- Corrected code\n"
+             "- Summary of changes")
+
+    uploaded_file = st.file_uploader("Upload your code file", type=SUPPORTED_EXTENSIONS)
+
+    if uploaded_file:
+        filename, code_text = read_code_file(uploaded_file)
+        st.code(code_text, language=filename.split(".")[-1])
+        
+        if st.button("🔍 Analyze and Fix Code"):
+            with st.spinner("Analyzing your code using Gemini..."):
+                prompt = (
+                    f"You are a software code analysis assistant. Given the code below, perform the following tasks:\n\n"
+                    f"1. Provide a short README-style summary of what the code is intended to do.\n"
+                    f"2. Identify any bugs or issues in the code.\n"
+                    f"3. Provide a corrected, clean version of the entire code.\n"
+                    f"4. List the changes you made in bullet points.\n\n"
+                    f"--- CODE START ---\n{code_text}\n--- CODE END ---"
+                )
 
                 response = client.models.generate_content(
                     model="gemini-2.5-flash",
-                    contents=content_blocks,
-                    config={"tools": [{"google_search": {}}]}
+                    contents=[{"text": prompt}]
                 )
 
-                st.success("📋 Analysis Result:")
+                st.success("✅ Analysis Complete:")
                 st.markdown(response.text)
 
 if __name__ == "__main__":
-    main()
+    main()