Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import google.generativeai as genai
-import requests
+import pyperclip
+import time
 
 # Configure the Gemini API
 genai.configure(api_key=st.secrets["GOOGLE_API_KEY"])
@@ -25,48 +26,58 @@ def generate_response(user_input):
     response = chat_session.send_message(user_input)
     return response.text
 
+def copy_to_clipboard(text):
+    pyperclip.copy(text)
+
 # Streamlit UI setup
-st.set_page_config(page_title="AI Code Assistant", page_icon="🤖", layout="wide")
+st.set_page_config(page_title="Advanced AI Code Assistant", page_icon="🚀", layout="wide")
 
 st.markdown("""
 <style>
-    @import url('https://fonts.googleapis.com/css2?family=Roboto:wght@300;400;700&display=swap');
+    @import url('https://fonts.googleapis.com/css2?family=Poppins:wght@300;400;600;700&display=swap');
     
     body {
-        font-family: 'Roboto', sans-serif;
-        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
-        color: #ffffff;
+        font-family: 'Poppins', sans-serif;
+        background: linear-gradient(120deg, #84fab0 0%, #8fd3f4 100%);
+        color: #2c3e50;
     }
     .stApp {
-        max-width: 1000px;
+        max-width: 1200px;
         margin: 0 auto;
         padding: 2rem;
-        background: rgba(255, 255, 255, 0.1);
+    }
+    .main-container {
+        background: rgba(255, 255, 255, 0.95);
         border-radius: 20px;
-        backdrop-filter: blur(10px);
+        padding: 2rem;
+        box-shadow: 0 10px 30px rgba(0, 0, 0, 0.1);
     }
     h1 {
-        font-size: 3rem;
+        font-size: 2.5rem;
         font-weight: 700;
+        color: #3498db;
+        text-align: center;
+        margin-bottom: 1rem;
+    }
+    .subtitle {
+        font-size: 1.2rem;
         text-align: center;
+        color: #7f8c8d;
         margin-bottom: 2rem;
-        text-shadow: 2px 2px 4px rgba(0,0,0,0.1);
     }
     .stTextArea label {
-        font-size: 1.2rem;
-        font-weight: 500;
-        color: #ffffff;
+        font-size: 1.1rem;
+        font-weight: 600;
+        color: #34495e;
     }
     .stTextArea textarea {
-        background: rgba(255, 255, 255, 0.2);
-        border: none;
+        border: 2px solid #3498db;
         border-radius: 10px;
-        color: #ffffff;
         font-size: 1rem;
-        padding: 1rem;
+        padding: 0.5rem;
     }
     .stButton button {
-        background: linear-gradient(45deg, #FF6B6B, #4ECDC4);
+        background: linear-gradient(45deg, #3498db, #2980b9);
         color: white;
         border: none;
         border-radius: 30px;
@@ -80,33 +91,66 @@ st.markdown("""
         transform: translateY(-2px);
         box-shadow: 0 6px 8px rgba(0,0,0,0.15);
     }
-    .stCodeBlock {
-        background-color: rgba(0, 0, 0, 0.6);
-        color: #f8f8f2;
+    .output-container {
+        background: #f8f9fa;
+        border-radius: 10px;
+        padding: 1rem;
+        margin-top: 2rem;
+    }
+    .code-block {
+        background-color: #282c34;
+        color: #abb2bf;
+        font-family: 'Fira Code', monospace;
         font-size: 0.9rem;
         border-radius: 10px;
         padding: 1rem;
         margin-top: 1rem;
-        overflow-x: auto;
+        position: relative;
+    }
+    .copy-btn {
+        position: absolute;
+        top: 0.5rem;
+        right: 0.5rem;
+        background: rgba(255, 255, 255, 0.1);
+        color: #abb2bf;
+        border: none;
+        border-radius: 5px;
+        padding: 0.2rem 0.5rem;
+        font-size: 0.8rem;
+        cursor: pointer;
+        transition: all 0.2s ease;
+    }
+    .copy-btn:hover {
+        background: rgba(255, 255, 255, 0.2);
     }
     .stAlert {
-        background-color: rgba(255, 255, 255, 0.2);
-        color: #ffffff;
+        background-color: #e8f4fd;
+        color: #3498db;
         border-radius: 10px;
         border: none;
+        padding: 0.5rem 1rem;
     }
     .stSpinner {
-        color: #ffffff;
+        color: #3498db;
     }
 </style>
 """, unsafe_allow_html=True)
 
-st.title("🤖 AI Code Assistant")
-st.markdown("<h3 style='text-align: center; color: #ffffff;'>Powered by Google Gemini</h3>", unsafe_allow_html=True)
+st.markdown('<div class="main-container">', unsafe_allow_html=True)
+st.title("🚀 Advanced AI Code Assistant")
+st.markdown('<p class="subtitle">Powered by Google Gemini</p>', unsafe_allow_html=True)
+
+prompt = st.text_area("What code can I help you with today?", height=100)
+
+col1, col2 = st.columns([1, 1])
 
-prompt = st.text_area("Enter your coding question or request:", height=150)
+with col1:
+    generate_button = st.button("Generate Code")
 
-if st.button("Generate Code"):
+with col2:
+    language = st.selectbox("Select language", ["Python", "JavaScript", "Java", "C++", "Ruby"])
+
+if generate_button:
     if prompt.strip() == "":
         st.error("Please enter a valid prompt.")
     else:
@@ -114,13 +158,22 @@ if st.button("Generate Code"):
             completed_text = generate_response(prompt)
             st.success("Code generated successfully!")
             
-            # Split the response into separate code blocks based on a custom delimiter
+            st.markdown('<div class="output-container">', unsafe_allow_html=True)
             code_blocks = completed_text.split("\n\n")
-            for block in code_blocks:
-                st.code(block, language="python")
+            for i, block in enumerate(code_blocks):
+                st.markdown(f'<div class="code-block">', unsafe_allow_html=True)
+                st.code(block, language=language.lower())
+                copy_button = st.button(f"Copy Code Block {i+1}", key=f"copy_{i}")
+                if copy_button:
+                    copy_to_clipboard(block)
+                    st.success(f"Code block {i+1} copied to clipboard!")
+                st.markdown('</div>', unsafe_allow_html=True)
+            st.markdown('</div>', unsafe_allow_html=True)
 
 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 style='text-align: center; margin-top: 2rem; color: #7f8c8d;'>
+    Created with ❤️ by Your Advanced AI Code Assistant
 </div>
-""", unsafe_allow_html=True)
+""", unsafe_allow_html=True)
+
+st.markdown('</div>', unsafe_allow_html=True)