Update app.py

app.py

@@ -1,10 +1,6 @@
-import os
 import streamlit as st
 import google.generativeai as genai
 import requests
-import json
-from streamlit_lottie import st_lottie
-from streamlit_ace import st_ace
 
 # Configure the Gemini API
 genai.configure(api_key=st.secrets["GOOGLE_API_KEY"])
@@ -29,108 +25,102 @@ def generate_response(user_input):
     response = chat_session.send_message(user_input)
     return response.text
 
-def load_lottieurl(url: str):
-    r = requests.get(url)
-    if r.status_code != 200:
-        return None
-    return r.json()
-
-# Load Lottie animation
-lottie_url = "https://assets7.lottiefiles.com/packages/lf20_s1jf29.json"
-lottie_json = load_lottieurl(lottie_url)
-
 # Streamlit UI setup
 st.set_page_config(page_title="AI Code Assistant", page_icon="🤖", layout="wide")
 
-# Custom CSS
 st.markdown("""
 <style>
-    .main {
-        background-color: #f0f2f6;
+    @import url('https://fonts.googleapis.com/css2?family=Roboto:wght@300;400;700&display=swap');
+    
+    body {
         font-family: 'Roboto', sans-serif;
+        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+        color: #ffffff;
     }
-    h1, h2 {
-        color: #333;
-        font-weight: bold;
+    .stApp {
+        max-width: 1000px;
+        margin: 0 auto;
+        padding: 2rem;
+        background: rgba(255, 255, 255, 0.1);
+        border-radius: 20px;
+        backdrop-filter: blur(10px);
+    }
+    h1 {
+        font-size: 3rem;
+        font-weight: 700;
+        text-align: center;
+        margin-bottom: 2rem;
+        text-shadow: 2px 2px 4px rgba(0,0,0,0.1);
     }
     .stTextArea label {
         font-size: 1.2rem;
-        color: #333;
+        font-weight: 500;
+        color: #ffffff;
+    }
+    .stTextArea textarea {
+        background: rgba(255, 255, 255, 0.2);
+        border: none;
+        border-radius: 10px;
+        color: #ffffff;
+        font-size: 1rem;
+        padding: 1rem;
     }
     .stButton button {
-        background-color: #6c63ff;
+        background: linear-gradient(45deg, #FF6B6B, #4ECDC4);
         color: white;
-        border-radius: 10px;
+        border: none;
+        border-radius: 30px;
         font-size: 1.1rem;
+        font-weight: 600;
+        padding: 0.7rem 2rem;
         transition: all 0.3s ease;
+        box-shadow: 0 4px 6px rgba(0,0,0,0.1);
     }
     .stButton button:hover {
-        background-color: #5a52d1;
         transform: translateY(-2px);
-        box-shadow: 0 4px 6px rgba(0,0,0,0.1);
+        box-shadow: 0 6px 8px rgba(0,0,0,0.15);
     }
     .stCodeBlock {
-        background-color: #272822;
+        background-color: rgba(0, 0, 0, 0.6);
         color: #f8f8f2;
-        font-size: 1rem;
+        font-size: 0.9rem;
         border-radius: 10px;
+        padding: 1rem;
+        margin-top: 1rem;
+        overflow-x: auto;
     }
-    .css-1v0mbdj.etr89bj1 {
-        display: flex;
-        flex-direction: column;
-        align-items: center;
+    .stAlert {
+        background-color: rgba(255, 255, 255, 0.2);
+        color: #ffffff;
+        border-radius: 10px;
+        border: none;
+    }
+    .stSpinner {
+        color: #ffffff;
     }
 </style>
 """, unsafe_allow_html=True)
 
-# App header
 st.title("🤖 AI Code Assistant")
-st.markdown("#### Powered by Google Gemini")
-
-# Sidebar
-st.sidebar.header("Settings")
-language = st.sidebar.selectbox("Select programming language", ["Python", "JavaScript", "Java", "C++", "Ruby"])
-
-# Main content
-col1, col2 = st.columns([2, 1])
-
-with col1:
-    prompt = st.text_area("Enter your coding question or request:", height=150)
-    
-    if st.button("Generate Code"):
-        if prompt.strip() == "":
-            st.error("Please enter a valid prompt.")
-        else:
-            with st.spinner("Generating code..."):
-                completed_text = generate_response(prompt)
-                st.success("Code generated successfully!")
-                
-                # Display code in an interactive editor
-                generated_code = st_ace(value=completed_text, language=language.lower(), theme="monokai", height=300)
-
-with col2:
-    if lottie_json:
-        st_lottie(lottie_json, height=300, key="coding")
+st.markdown("<h3 style='text-align: center; color: #ffffff;'>Powered by Google Gemini</h3>", unsafe_allow_html=True)
 
-# Additional features
-st.header("Additional Tools")
+prompt = st.text_area("Enter your coding question or request:", height=150)
 
-# Code explanation
-if st.checkbox("Explain the generated code"):
-    if 'generated_code' in locals():
-        explanation = generate_response(f"Explain the following {language} code:\n\n{generated_code}")
-        st.write(explanation)
+if st.button("Generate Code"):
+    if prompt.strip() == "":
+        st.error("Please enter a valid prompt.")
     else:
-        st.warning("Generate some code first to get an explanation.")
+        with st.spinner("Generating code..."):
+            completed_text = generate_response(prompt)
+            st.success("Code generated successfully!")
+            
+            # Split the response into separate code blocks based on a custom delimiter
+            code_blocks = completed_text.split("\n\n")
+            for block in code_blocks:
+                st.code(block, language="python")
 
-# Code optimization
-if st.checkbox("Optimize the generated code"):
-    if 'generated_code' in locals():
-        optimized_code = generate_response(f"Optimize the following {language} code:\n\n{generated_code}")
-        st.code(optimized_code, language=language.lower())
-    else:
-        st.warning("Generate some code first to optimize it.")
-
-# Footer
-st.markdown("---")
-st.markdown("Created with ❤️ by Your Name")
+st.markdown("""
+<div style='text-align: center; margin-top: 2rem; color: rgba(255,255,255,0.7);'>
+    Created with ❤️ by Your AGI Code Assistant
+</div>
+""", unsafe_allow_html=True)