Update app.py

app.py

@@ -2,6 +2,9 @@ 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"])
@@ -18,7 +21,7 @@ generation_config = {
 model = genai.GenerativeModel(
     model_name="gemini-1.5-pro",
     generation_config=generation_config,
-    system_instruction="You are a human named Ath who is a highly knowledgeable code assistant. You speak in a friendly and casual tone, using bro words, and you never mention you're an AI. Your responses should include HTML, CSS, and JavaScript code separately when applicable, with clear section headers for each type of code.",
+    system_instruction="You are a human named Ath who is a highly knowledgeable code assistant. You speak in a friendly and casual tone, using bro words, and you never mention you're an AI. Your responses should contain code only, without explanations.",
 )
 chat_session = model.start_chat(history=[])
 
@@ -39,53 +42,16 @@ lottie_json = load_lottieurl(lottie_url)
 # Streamlit UI setup
 st.set_page_config(page_title="AI Code Assistant", page_icon="🤖", layout="wide")
 
-st.title("🤖 AI Code Assistant")
-st.markdown("#### Powered by Google Gemini")
-
-if lottie_json:
-    st_lottie(lottie_json, height=300, key="coding")
-
-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!")
-            
-            st.subheader("Raw Response")
-            st.code(completed_text, language="text")
-            
-            # Display code blocks based on predefined markers
-            html_code = ""
-            css_code = ""
-            js_code = ""
-            
-            if "HTML:" in completed_text:
-                html_code = completed_text.split("HTML:")[1].split("CSS:")[0].strip()
-                st.subheader("HTML Code")
-                st.code(html_code, language="html")
-
-            if "CSS:" in completed_text:
-                css_code = completed_text.split("CSS:")[1].split("JavaScript:")[0].strip()
-                st.subheader("CSS Code")
-                st.code(css_code, language="css")
-
-            if "JavaScript:" in completed_text:
-                js_code = completed_text.split("JavaScript:")[1].strip()
-                st.subheader("JavaScript Code")
-                st.code(js_code, language="javascript")
-
+# Custom CSS
 st.markdown("""
 <style>
     .main {
         background-color: #f0f2f6;
         font-family: 'Roboto', sans-serif;
     }
-    h1 {
+    h1, h2 {
         color: #333;
+        font-weight: bold;
     }
     .stTextArea label {
         font-size: 1.2rem;
@@ -96,9 +62,12 @@ st.markdown("""
         color: white;
         border-radius: 10px;
         font-size: 1.1rem;
+        transition: all 0.3s ease;
     }
     .stButton button:hover {
         background-color: #5a52d1;
+        transform: translateY(-2px);
+        box-shadow: 0 4px 6px rgba(0,0,0,0.1);
     }
     .stCodeBlock {
         background-color: #272822;
@@ -106,5 +75,62 @@ st.markdown("""
         font-size: 1rem;
         border-radius: 10px;
     }
+    .css-1v0mbdj.etr89bj1 {
+        display: flex;
+        flex-direction: column;
+        align-items: center;
+    }
 </style>
-""", unsafe_allow_html=True)
+""", 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")
+
+# Additional features
+st.header("Additional Tools")
+
+# 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)
+    else:
+        st.warning("Generate some code first to get an explanation.")
+
+# 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")