Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st 
 import time
+import google.generativeai as genai
 
 # Animated text function
 def animated_text(text, speed=0.05):
@@ -9,14 +10,12 @@ def animated_text(text, speed=0.05):
         displayed_text += letter
         placeholder.markdown(f"""
             <h1 style="text-align:center; color: #FF4500;">{displayed_text} 🚀</h1>
-        """, unsafe_allow_html=True)  # Corrected f-string formatting
+        """, unsafe_allow_html=True)
         time.sleep(speed)
 
 # Display animated text
 animated_text("Welcome to GenAI App - AI Code Reviewer!", speed=0.1)
 
-import google.generativeai as genai
-
 # Set up Google Gemini API Key
 genai.configure(api_key="AIzaSyBm0GOvYox4OyRG1WFOK7FT5fnNCHfubns")
 
@@ -29,41 +28,40 @@ buggy_code = st.text_area("🐞 Enter your buggy code here:", height=200)
 # 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)
-
-    # Display Fixed Code
-    st.subheader("✅ Fixed Code:")
-    st.code(fixed_code, language="python")  # Adjust language if needed
-
-    # Copy Button for Fixed Code
-    st.download_button(
-        label="📋 Copy Fixed Code",
-        data=fixed_code,
-        file_name="fixed_code.py",
-        mime="text/plain"
-    )
-
-    # Display AI-Generated Suggestions
-    st.subheader("💡 Suggestions for Improvement:")
-    st.write(suggestions)
+# Generate Button
+if st.button("🚀 Analyze & Optimize Code"):
+    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)
+        
+        # Display Fixed Code
+        st.subheader("✅ Fixed Code:")
+        st.code(fixed_code, language="python")
+        
+        # Copy Button for Fixed Code
+        st.download_button(
+            label="📋 Copy Fixed Code",
+            data=fixed_code,
+            file_name="fixed_code.py",
+            mime="text/plain"
+        )
+        
+        # Display AI-Generated Suggestions
+        st.subheader("💡 Suggestions for Improvement:")
+        st.write(suggestions)
+    else:
+        st.warning("⚠️ Please enter some code before generating fixes and suggestions!")