Update app.py

app.py

@@ -1,5 +1,9 @@
 import streamlit as st
 import google.generativeai as genai
+from pygments import highlight
+from pygments.lexers import get_lexer_by_name
+from pygments.formatters import HtmlFormatter
+import html
 
 # Configure the Gemini API
 genai.configure(api_key=st.secrets["GOOGLE_API_KEY"])
@@ -26,180 +30,170 @@ def generate_response(user_input):
     except Exception as e:
         return f"An error occurred: {e}"
 
+def create_code_block(code, language):
+    lexer = get_lexer_by_name(language, stripall=True)
+    formatter = HtmlFormatter(style="monokai", linenos=True, cssclass="source")
+    highlighted_code = highlight(code, lexer, formatter)
+    css = formatter.get_style_defs('.source')
+    return highlighted_code, css
+
 # Streamlit UI setup
-st.set_page_config(page_title="CodeGenius", 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=Inter:wght@300;400;600&family=JetBrains+Mono:wght@400;500&display=swap');
+    @import url('https://fonts.googleapis.com/css2?family=Poppins:wght@300;400;600;700&display=swap');
     
     body {
-        font-family: 'Inter', sans-serif;
-        background-color: #f8f9fa;
-        color: #343a40;
+        font-family: 'Poppins', sans-serif;
+        background-color: #f0f7ff;
+        color: #333;
     }
     .stApp {
-        max-width: 900px;
+        max-width: 1200px;
         margin: 0 auto;
         padding: 2rem;
     }
     .main-container {
         background: #ffffff;
-        border-radius: 12px;
-        padding: 2rem;
-        box-shadow: 0 4px 6px rgba(0, 0, 0, 0.05);
+        border-radius: 20px;
+        padding: 3rem;
+        box-shadow: 0 10px 30px rgba(0, 0, 0, 0.1);
     }
     h1 {
-        font-size: 2.5rem;
-        font-weight: 600;
-        color: #212529;
+        font-size: 3rem;
+        font-weight: 700;
+        color: #1e3a8a;
         text-align: center;
-        margin-bottom: 0.5rem;
+        margin-bottom: 1rem;
+        text-shadow: 2px 2px 4px rgba(0,0,0,0.1);
     }
     .subtitle {
-        font-size: 1rem;
+        font-size: 1.2rem;
         text-align: center;
-        color: #6c757d;
-        margin-bottom: 2rem;
+        color: #64748b;
+        margin-bottom: 3rem;
     }
     .stTextArea textarea {
-        background-color: #f1f3f5;
-        color: #495057;
-        border: 1px solid #ced4da;
-        border-radius: 6px;
+        border: 2px solid #cbd5e1;
+        border-radius: 12px;
         font-size: 1rem;
-        font-family: 'JetBrains Mono', monospace;
-        padding: 0.75rem;
-        transition: all 0.2s ease;
+        padding: 1rem;
+        transition: all 0.3s ease;
+        box-shadow: inset 0 2px 4px rgba(0,0,0,0.05);
     }
     .stTextArea textarea:focus {
-        border-color: #4dabf7;
-        box-shadow: 0 0 0 2px rgba(77, 171, 247, 0.2);
+        border-color: #3b82f6;
+        box-shadow: 0 0 0 3px rgba(59, 130, 246, 0.5);
     }
     .stButton button {
-        background-color: #4dabf7;
-        color: #ffffff;
+        background-color: #3b82f6;
+        color: white;
         border: none;
-        border-radius: 6px;
-        font-size: 1rem;
-        font-weight: 500;
-        padding: 0.6rem 1.2rem;
-        transition: all 0.2s ease;
+        border-radius: 12px;
+        font-size: 1.1rem;
+        font-weight: 600;
+        padding: 0.75rem 2rem;
+        transition: all 0.3s ease;
+        width: 100%;
+        box-shadow: 0 4px 6px rgba(59, 130, 246, 0.3);
     }
     .stButton button:hover {
-        background-color: #3793dd;
-        transform: translateY(-1px);
+        background-color: #2563eb;
+        transform: translateY(-2px);
+        box-shadow: 0 6px 8px rgba(59, 130, 246, 0.4);
     }
     .output-container {
-        background: #f8f9fa;
-        border-radius: 6px;
+        background: #f8fafc;
+        border-radius: 12px;
         padding: 1.5rem;
         margin-top: 2rem;
-        border: 1px solid #e9ecef;
+        border: 1px solid #e2e8f0;
     }
     .code-block {
-        background-color: #f1f3f5;
-        color: #212529;
-        font-family: 'JetBrains Mono', monospace;
-        font-size: 0.9rem;
-        border-radius: 6px;
-        padding: 1.25rem;
+        background-color: #272822;
+        border-radius: 12px;
+        padding: 1.5rem;
         margin-top: 1rem;
         overflow-x: auto;
     }
     .stAlert {
-        background-color: #e9ecef;
-        color: #495057;
-        border-radius: 6px;
-        border: 1px solid #ced4da;
-        padding: 0.75rem 1rem;
+        background-color: #dbeafe;
+        color: #1e40af;
+        border-radius: 12px;
+        border: none;
+        padding: 1rem 1.5rem;
         margin-bottom: 1rem;
     }
     .stSpinner {
-        color: #4dabf7;
+        color: #3b82f6;
     }
     /* Custom scrollbar */
     ::-webkit-scrollbar {
-        width: 6px;
-        height: 6px;
+        width: 8px;
+        height: 8px;
     }
     ::-webkit-scrollbar-track {
-        background: #f1f3f5;
-        border-radius: 3px;
+        background: #f1f5f9;
+        border-radius: 4px;
     }
     ::-webkit-scrollbar-thumb {
-        background: #adb5bd;
-        border-radius: 3px;
+        background: #94a3b8;
+        border-radius: 4px;
     }
     ::-webkit-scrollbar-thumb:hover {
-        background: #868e96;
+        background: #64748b;
     }
-    .language-selector {
-        margin-bottom: 1rem;
-    }
-    .copy-button {
-        background-color: #e9ecef;
-        color: #495057;
-        border: none;
-        border-radius: 4px;
-        padding: 0.4rem 0.8rem;
+    .source {
+        font-family: 'Fira Code', monospace;
         font-size: 0.9rem;
-        cursor: pointer;
-        transition: all 0.2s ease;
+        line-height: 1.4;
+    }
+    .source .linenos {
+        color: #75715e;
+        padding-right: 10px;
+        border-right: 1px solid #49483e;
+        user-select: none;
     }
-    .copy-button:hover {
-        background-color: #ced4da;
+    .source pre {
+        margin: 0;
+        padding: 0;
     }
 </style>
 """, unsafe_allow_html=True)
 
 st.markdown('<div class="main-container">', unsafe_allow_html=True)
-st.title("🚀 CodeGenius")
-st.markdown('<p class="subtitle">Elevate Your Code with AI-Powered Insights</p>', unsafe_allow_html=True)
-
-languages = ["Python", "JavaScript", "Java", "C++", "Ruby", "Go", "Rust", "TypeScript", "PHP", "Swift"]
-selected_language = st.selectbox("Select your programming language:", languages, key="language-selector")
-
-prompt = st.text_area(f"What {selected_language} challenge can I help you with?", height=100)
+st.title("💻 Advanced AI Code Assistant")
+st.markdown('<p class="subtitle">Powered by Google Gemini - Expert-level coding solutions</p>', unsafe_allow_html=True)
 
-col1, col2, col3 = st.columns([2, 1, 1])
-with col1:
-    generate_button = st.button("Generate Solution")
-with col2:
-    complexity = st.radio("Complexity", ["Low", "Medium", "High"], horizontal=True)
-with col3:
-    style = st.radio("Style", ["Concise", "Detailed"], horizontal=True)
+prompt = st.text_area("What advanced coding challenge can I assist you with today?", height=120)
 
-if generate_button:
+if st.button("Generate Expert Code"):
     if prompt.strip() == "":
-        st.warning("Please enter a valid prompt.")
+        st.error("Please enter a valid prompt.")
     else:
-        with st.spinner(f"Crafting {complexity.lower()} complexity {selected_language} solution..."):
-            completed_text = generate_response(f"{complexity} complexity, {style} {selected_language} code for: {prompt}")
+        with st.spinner("Generating advanced code solution..."):
+            completed_text = generate_response(prompt)
             if "An error occurred" in completed_text:
                 st.error(completed_text)
             else:
-                st.success(f"Solution generated successfully!")
+                st.success("Expert-level code generated successfully!")
                 
+                # Attempt to determine the language (this is a simple guess, you might want to improve this)
+                language = "python" if "def " in completed_text or "import " in completed_text else "javascript"
+                
+                highlighted_code, css = create_code_block(completed_text, language)
+                
+                st.markdown(f'<style>{css}</style>', unsafe_allow_html=True)
                 st.markdown('<div class="output-container">', unsafe_allow_html=True)
                 st.markdown('<div class="code-block">', unsafe_allow_html=True)
-                st.code(completed_text, language=selected_language.lower())
+                st.markdown(highlighted_code, unsafe_allow_html=True)
                 st.markdown('</div>', unsafe_allow_html=True)
-                
-                # Add copy to clipboard button
-                st.markdown(
-                    f"""
-                    <button onclick="navigator.clipboard.writeText(`{completed_text}`)" class="copy-button">
-                        Copy to Clipboard
-                    </button>
-                    """,
-                    unsafe_allow_html=True
-                )
                 st.markdown('</div>', unsafe_allow_html=True)
 
 st.markdown("""
-<div style='text-align: center; margin-top: 2rem; color: #adb5bd; font-size: 0.9rem;'>
-    Powered by CodeGenius AI • Simplifying Complex Code Challenges
+<div style='text-align: center; margin-top: 2rem; color: #4a5568;'>
+    Crafted with ❤️ by Your Advanced AI Code Assistant
 </div>
 """, unsafe_allow_html=True)