Update app.py

app.py

@@ -16,28 +16,54 @@ def animated_text(text, speed=0.05):
 animated_text("Welcome to GenAI App - AI Code Reviewer!", speed=0.1)
 
 import google.generativeai as genai
-st.header('🚀 AI-Powered Code Debugging 🔍: ')
 
-st.subheader('🐞 Buggy Code::')
+# Set up Google Gemini API Key
+genai.configure(api_key=st.secrets["AIzaSyBm0GOvYox4OyRG1WFOK7FT5fnNCHfubns"])
 
-# Text Area for User Input
-code = st.text_area("Enter your code here:", height=200)
+# Streamlit App Title
+st.title("🤖 AI-Powered Code Debugging")
 
-# Display the Code with Syntax Highlighting
-if code:
-    st.code(code, language="python")  # Adjust language as needed
+# Text Area for User Input (Buggy Code)
+buggy_code = st.text_area("🐞 Enter your buggy code here:", height=200)
 
-    # Copy Code Button
-    st.markdown(
-        f"""
-        <button onclick="navigator.clipboard.writeText(`{code}`)">Debug Code</button>
-        """,
-        unsafe_allow_html=True
-    )
+# Function to Debug Code Using Gemini AI
+def debug_code_with_gemini(code):
+    prompt = f"Debug the following code and provide the corrected version:\n\n{code}"
+    
+    model = genai.GenerativeModel("models/gemini-2.0-flash")
+    response = model.generate_content(prompt)
+    
+    return response.text if response.text else "⚠️ Error in AI response."
+
+# Function to Generate Code Improvement Suggestions
+def get_suggestions_with_gemini(code):
+    prompt = f"Suggest improvements for the following code, focusing on best practices, performance optimization, and readability:\n\n{code}"
+    
+    model = genai.GenerativeModel("gemini-pro")
+    response = model.generate_content(prompt)
+    
+    return response.text if response.text else "⚠️ No suggestions available."
 
+# Process Code When Input is Given
+if buggy_code:
+    with st.spinner("🛠️ Debugging your code..."):
+        fixed_code = debug_code_with_gemini(buggy_code)
+    
+    with st.spinner("🔍 Generating suggestions..."):
+        suggestions = get_suggestions_with_gemini(fixed_code)
 
-st.subheader('🔍 Bug Found in:')
+    # Display Fixed Code
+    st.subheader("✅ Fixed Code:")
+    st.code(fixed_code, language="python")  # Adjust language if needed
 
-st.subheader('✅ Fixed Code:')
+    # Copy Button for Fixed Code
+    st.download_button(
+        label="📋 Copy Fixed Code",
+        data=fixed_code,
+        file_name="fixed_code.py",
+        mime="text/plain"
+    )
 
-st.subheader('💡 Suggestion:')
+    # Display AI-Generated Suggestions
+    st.subheader("💡 Suggestions for Improvement:")
+    st.write(suggestions)