Update app.py

app.py

@@ -1,69 +1,76 @@
-import streamlit as st 
+import streamlit as st
 import time
+import google.generativeai as genai
+from PIL import Image
+import base64
+
+# Configure Google Gemini API Key
+genai.configure(api_key="AIzaSyC7op0zN_EESTSFhXGtzHRYImp5nhvDz-c")
+
+# Custom CSS for advanced styling
+st.markdown("""
+    <style>
+        body {background-color: #0E0E0E; color: #FFFFFF;}
+        .main-title {text-align: center; font-size: 40px; font-weight: bold; color: #FFA500;}
+        .sub-header {text-align: center; font-size: 25px; font-weight: bold; color: #FFD700;}
+        .stTextArea textarea {background-color: #222; color: #FFA500; border-radius: 10px;}
+        .stButton button {background-color: #FF4500; color: white; font-size: 18px; border-radius: 10px;}
+        .stDownloadButton button {background-color: #32CD32; color: white; font-size: 18px; border-radius: 10px;}
+    </style>
+""", unsafe_allow_html=True)
 
-# Animated text function
-def animated_text(text, speed=0.05):
+# Function to display animated text
+def animated_text(text, speed=0.07):
     placeholder = st.empty()
     displayed_text = ""
     for letter in text:
         displayed_text += letter
-        placeholder.markdown(f"""
-            <h1 style="text-align:center; color: #FF4500;">{displayed_text} 🚀</h1>
-        """, unsafe_allow_html=True)  # Corrected f-string formatting
+        placeholder.markdown(f'<h1 class="main-title">{displayed_text} 🚀</h1>', 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="AIzaSyC7op0zN_EESTSFhXGtzHRYImp5nhvDz-c")
+# Animated Welcome Text
+animated_text("Welcome to AI Code Debugger & Optimizer!")
 
-# Streamlit App Title
-st.title("🤖 AI-Powered Code Debugging")
+# Streamlit Layout
+st.markdown("<h2 class='sub-header'>🚀 AI-Powered Code Debugging & Enhancement</h2>", unsafe_allow_html=True)
 
-# Text Area for User Input (Buggy Code)
-buggy_code = st.text_area("🐞 Enter your buggy code here:", height=200)
+# User Input for Buggy Code
+buggy_code = st.text_area("🐞 Paste your buggy code below:", height=200)
 
-# Function to Debug Code Using Gemini AI
+# Function to debug code
 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
+# Function to get suggestions for improvement
 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
+# Processing user input
 if buggy_code:
     with st.spinner("🛠️ Debugging your code..."):
         fixed_code = debug_code_with_gemini(buggy_code)
     
-    with st.spinner("🔍 Generating suggestions..."):
+    with st.spinner("🔍 Generating optimization 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
+    # Display Debugged Code
+    st.markdown("<h2 class='sub-header'>✅ Fixed Code:</h2>", unsafe_allow_html=True)
+    st.code(fixed_code, language="python")
+    
+    # Download button for Fixed Code
     st.download_button(
-        label="📋 Copy Fixed Code",
+        label="📥 Download 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)
+    
+    # Display AI Suggestions
+    st.markdown("<h2 class='sub-header'>💡 AI-Generated Suggestions:</h2>", unsafe_allow_html=True)
+    st.write(suggestions)