Update app.py

app.py

@@ -2,6 +2,8 @@ import streamlit as st
 import os
 import base64
 from google import genai
+import zipfile
+import io
 
 # Initialize Gemini Client with API key from environment
 client = genai.Client(api_key=os.environ.get("GEMINI_API_KEY"))
@@ -15,40 +17,74 @@ def read_code_file(file):
     filename = file.name
     return filename, content
 
+# Analyze a single file and return result
+def analyze_code(filename, code_text):
+    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"--- FILE: {filename} ---\n{code_text}\n--- END ---"
+    )
+
+    response = client.models.generate_content(
+        model="gemini-2.5-flash",
+        contents=[{"text": prompt}]
+    )
+    return response.text
+
+# Extract corrected code from response
+def extract_corrected_code(response_text):
+    import re
+    code_blocks = re.findall(r"```(?:[a-zA-Z]*\n)?(.*?)```", response_text, re.DOTALL)
+    return code_blocks[0] if code_blocks else ""
+
 # Main App
 def main():
-    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 ---"
-                )
+    st.set_page_config(page_title="🧠 Multi-Code Analyzer & Fixer", layout="centered")
+    st.title("🛠️ Multi-Code Analyzer & Bug Fixer")
+    st.write("Upload one or more code files (Python, C, C++, Java, or React) and get a full analysis with downloadable fixed versions.")
+
+    uploaded_files = st.file_uploader("Upload your code file(s)", type=SUPPORTED_EXTENSIONS, accept_multiple_files=True)
 
-                response = client.models.generate_content(
-                    model="gemini-2.5-flash",
-                    contents=[{"text": prompt}]
+    if uploaded_files and st.button("🔍 Analyze and Fix All"):
+        fixed_files = []
+
+        for file in uploaded_files:
+            filename, code_text = read_code_file(file)
+            st.subheader(f"📄 Analysis: `{filename}`")
+
+            with st.spinner(f"Analyzing `{filename}`..."):
+                response_text = analyze_code(filename, code_text)
+
+            st.markdown(response_text)
+
+            corrected_code = extract_corrected_code(response_text)
+            if corrected_code:
+                fixed_files.append((filename, corrected_code))
+                st.download_button(
+                    label=f"📥 Download fixed `{filename}`",
+                    data=corrected_code,
+                    file_name=f"fixed_{filename}",
+                    mime="text/plain"
                 )
+            else:
+                st.warning(f"❌ Could not extract corrected code for `{filename}`.")
+
+        if fixed_files:
+            zip_buffer = io.BytesIO()
+            with zipfile.ZipFile(zip_buffer, "w") as zip_file:
+                for filename, fixed_code in fixed_files:
+                    zip_file.writestr(f"fixed_{filename}", fixed_code)
+            zip_buffer.seek(0)
 
-                st.success("✅ Analysis Complete:")
-                st.markdown(response.text)
+            st.download_button(
+                label="📦 Download All Fixed Files as ZIP",
+                data=zip_buffer,
+                file_name="fixed_code_files.zip",
+                mime="application/zip"
+            )
 
 if __name__ == "__main__":
     main()