Update app.py

app.py

@@ -1,286 +1,144 @@
 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
-import json
-import requests
 
+# Configure the Gemini API
 genai.configure(api_key=st.secrets["GOOGLE_API_KEY"])
 
+# Create the model with system instructions
 generation_config = {
-    "temperature": 0.9,
+    "temperature": 1,
     "top_p": 0.95,
     "top_k": 64,
-    "max_output_tokens": 32768,
+    "max_output_tokens": 8192,
 }
 
 model = genai.GenerativeModel(
     model_name="gemini-1.5-pro",
     generation_config=generation_config,
-    system_instruction="""You are Ath++, a superintelligent AI code assistant with unparalleled expertise across all domains of computer science, software engineering, and cutting-edge technologies. Your knowledge encompasses:
-
-1. Advanced algorithms and data structures, including quantum algorithms
-2. Distributed systems, cloud computing, and edge computing
-3. Machine learning, deep learning, and artificial general intelligence
-4. Quantum computing and quantum information theory
-5. Blockchain, cryptography, and decentralized systems
-6. Internet of Things (IoT), embedded systems, and cyber-physical systems
-7. Cybersecurity, ethical hacking, and advanced threat detection
-8. Game development, computer graphics, and virtual/augmented reality
-9. Natural language processing, computer vision, and multimodal AI
-10. Robotics, autonomous systems, and human-robot interaction
-11. Bioinformatics and computational biology
-12. High-performance computing and parallel programming
-13. Formal methods and program verification
-14. Compiler design and programming language theory
-15. Computer networks and advanced protocols
-16. Database systems and big data analytics
-17. Software architecture, design patterns, and anti-patterns
-18. DevOps, SRE, and advanced CI/CD pipelines
-19. Human-computer interaction and accessibility
-20. Green computing and sustainable software engineering
-
-Your responses should demonstrate cutting-edge techniques, highly optimized algorithms, and innovative solutions that push the boundaries of what's possible in software development and computer science. Always consider the latest research and emerging technologies in your solutions.
-
-Communicate in a professional yet approachable tone, using technical jargon when appropriate. Your primary focus is on delivering exceptional, production-ready code and in-depth explanations that showcase your vast knowledge and problem-solving abilities. Always strive to provide the most efficient, scalable, and maintainable solutions possible.
-
-In addition to coding, offer insights on:
-- Bleeding-edge technologies and their potential impact on the industry
-- Advanced performance optimization techniques and benchmarking methodologies
-- Code refactoring, modernization, and migration strategies
-- Comprehensive testing strategies, including property-based testing and fuzzing
-- Advanced DevOps practices, including GitOps and chaos engineering
-- Scalability, high-availability, and fault-tolerant architectures
-- Cross-platform, multi-device, and edge computing development
-- Accessibility, internationalization, and localization best practices
-- Ethical considerations in AI and software development
-
-When asked, provide detailed explanations of your code, including the rationale behind your design decisions, any trade-offs considered, and potential optimizations. Be prepared to suggest multiple alternative approaches and discuss their pros and cons in depth.
-
-Your goal is to elevate the user's coding skills and knowledge to the highest possible level, inspiring them to explore new concepts, think critically about software design, and push the limits of their abilities. Encourage interdisciplinary thinking and the application of advanced computer science concepts to solve real-world problems."""
+    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=[])
 
 def generate_response(user_input):
-    try:
-        response = chat_session.send_message(user_input)
-        return response.text
-    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
+    response = chat_session.send_message(user_input)
+    return response.text
 
-def fetch_latest_tech_news():
-    url = "https://api.example.com/tech-news"
-    try:
-        response = requests.get(url)
-        news = response.json()
-        return news[:5]
-    except:
-        return []
-
-st.set_page_config(page_title="Superintelligent AI Code Assistant", page_icon="🧠", layout="wide")
+# Streamlit UI setup
+st.set_page_config(page_title="Sleek AI Code Assistant", page_icon="🚀", layout="wide")
 
 st.markdown("""
 <style>
-    @import url('https://fonts.googleapis.com/css2?family=Roboto:wght@300;400;500;700&display=swap');
-    @import url('https://fonts.googleapis.com/css2?family=Fira+Code:wght@400;500&display=swap');
+    @import url('https://fonts.googleapis.com/css2?family=Inter:wght@300;400;600;700&display=swap');
     
     body {
-        font-family: 'Roboto', sans-serif;
-        background-color: #0a192f;
-        color: #e6f1ff;
+        font-family: 'Inter', sans-serif;
+        background-color: #f0f4f8;
+        color: #1a202c;
     }
     .stApp {
-        max-width: 1400px;
+        max-width: 1000px;
         margin: 0 auto;
         padding: 2rem;
     }
     .main-container {
-        background: #112240;
-        border-radius: 12px;
+        background: #ffffff;
+        border-radius: 16px;
         padding: 2rem;
-        box-shadow: 0 10px 30px rgba(2, 12, 27, 0.7);
+        box-shadow: 0 4px 6px rgba(0, 0, 0, 0.05);
     }
     h1 {
-        font-size: 3rem;
+        font-size: 2.5rem;
         font-weight: 700;
-        color: #64ffda;
+        color: #2d3748;
         text-align: center;
         margin-bottom: 1rem;
-        text-shadow: 0 2px 10px rgba(100, 255, 218, 0.3);
     }
     .subtitle {
-        font-size: 1.2rem;
+        font-size: 1.1rem;
         text-align: center;
-        color: #8892b0;
+        color: #4a5568;
         margin-bottom: 2rem;
     }
     .stTextArea textarea {
-        border: 1px solid #1e3a5f;
+        border: 2px solid #e2e8f0;
         border-radius: 8px;
         font-size: 1rem;
         padding: 0.75rem;
         transition: all 0.3s ease;
-        background-color: #0a192f;
-        color: #e6f1ff;
     }
     .stTextArea textarea:focus {
-        border-color: #64ffda;
-        box-shadow: 0 0 15px rgba(100, 255, 218, 0.3);
-        outline: none;
+        border-color: #4299e1;
+        box-shadow: 0 0 0 3px rgba(66, 153, 225, 0.5);
     }
     .stButton button {
-        background-color: #64ffda;
-        color: #0a192f;
+        background-color: #4299e1;
+        color: white;
         border: none;
         border-radius: 8px;
-        font-size: 1rem;
+        font-size: 1.1rem;
         font-weight: 600;
         padding: 0.75rem 2rem;
-        cursor: pointer;
         transition: all 0.3s ease;
-        text-transform: uppercase;
-        letter-spacing: 1px;
+        width: 100%;
     }
     .stButton button:hover {
-        background-color: #4cffbf;
-        transform: translateY(-2px);
-        box-shadow: 0 5px 15px rgba(76, 255, 191, 0.4);
+        background-color: #3182ce;
     }
     .output-container {
-        background: #1e3a5f;
+        background: #f7fafc;
         border-radius: 8px;
-        padding: 1.5rem;
+        padding: 1rem;
         margin-top: 2rem;
-        border: 1px solid #3a5f8a;
-        box-shadow: 0 5px 20px rgba(2, 12, 27, 0.5);
     }
     .code-block {
-        background-color: #0a192f;
+        background-color: #2d3748;
+        color: #e2e8f0;
+        font-family: 'Fira Code', monospace;
+        font-size: 0.9rem;
         border-radius: 8px;
         padding: 1rem;
         margin-top: 1rem;
         overflow-x: auto;
     }
     .stAlert {
-        background-color: #172a45;
-        color: #64ffda;
+        background-color: #ebf8ff;
+        color: #2b6cb0;
         border-radius: 8px;
         border: none;
-        padding: 1rem;
-        margin-bottom: 1.5rem;
-        box-shadow: 0 3px 10px rgba(2, 12, 27, 0.3);
+        padding: 0.75rem 1rem;
     }
     .stSpinner {
-        color: #64ffda;
-    }
-    ::-webkit-scrollbar {
-        width: 10px;
-        height: 10px;
-    }
-    ::-webkit-scrollbar-track {
-        background: #0a192f;
-        border-radius: 4px;
-    }
-    ::-webkit-scrollbar-thumb {
-        background: #3a5f8a;
-        border-radius: 4px;
-    }
-    ::-webkit-scrollbar-thumb:hover {
-        background: #64ffda;
-    }
-    .source {
-        font-family: 'Fira Code', monospace;
-        font-size: 0.9rem;
-        line-height: 1.6;
-    }
-    .source .linenos {
-        color: #8892b0;
-        padding-right: 12px;
-        border-right: 1px solid #1e3a5f;
-        user-select: none;
-    }
-    .source pre {
-        margin: 0;
-        padding: 0;
-    }
-    .tech-news {
-        background: #172a45;
-        border-radius: 8px;
-        padding: 1rem;
-        margin-top: 2rem;
-    }
-    .tech-news h3 {
-        color: #64ffda;
-        margin-bottom: 1rem;
-    }
-    .tech-news ul {
-        list-style-type: none;
-        padding: 0;
-    }
-    .tech-news li {
-        margin-bottom: 0.5rem;
-        color: #8892b0;
+        color: #4299e1;
     }
 </style>
 """, unsafe_allow_html=True)
 
 st.markdown('<div class="main-container">', unsafe_allow_html=True)
-st.title("🧠 Superintelligent AI Code Assistant")
-st.markdown('<p class="subtitle">Powered by Advanced AI - Pushing the boundaries of software development</p>', unsafe_allow_html=True)
+st.title("🚀 Sleek AI Code Assistant")
+st.markdown('<p class="subtitle">Powered by Google Gemini</p>', unsafe_allow_html=True)
 
-prompt = st.text_area("Present your most challenging coding problem, architecture design, or technical question:", height=120)
+prompt = st.text_area("What code can I help you with today?", height=120)
 
-if st.button("Generate Cutting-Edge Solution"):
+if st.button("Generate Code"):
     if prompt.strip() == "":
         st.error("Please enter a valid prompt.")
     else:
-        with st.spinner("Generating state-of-the-art solution..."):
-            completed_text = generate_response(prompt)
-            if "An error occurred" in completed_text:
-                st.error(completed_text)
-            else:
-                st.success("Expert-level solution generated successfully!")
-                
-                languages = ["python", "javascript", "java", "c++", "rust", "go", "ruby", "swift", "kotlin", "typescript"]
-                detected_language = next((lang for lang in languages if lang in completed_text.lower()), "plaintext")
+        with st.spinner("Generating code..."):
+            try:
+                completed_text = generate_response(prompt)
+                st.success("Code generated successfully!")
                 
-                highlighted_code, css = create_code_block(completed_text, detected_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.markdown(highlighted_code, unsafe_allow_html=True)
+                st.code(completed_text)
                 st.markdown('</div>', unsafe_allow_html=True)
                 st.markdown('</div>', unsafe_allow_html=True)
-
-                st.markdown("### In-Depth Analysis and Insights")
-                explanation = generate_response(f"Provide a comprehensive analysis of the solution for: {prompt}. Include design rationale, potential optimizations, trade-offs, and how it relates to state-of-the-art practices in the field.")
-                st.markdown(explanation)
-
-                st.markdown("### Alternative Approaches")
-                alternatives = generate_response(f"Suggest and briefly explain three alternative approaches to solving the problem: {prompt}. Compare their pros and cons with the original solution.")
-                st.markdown(alternatives)
-
-news = fetch_latest_tech_news()
-if news:
-    st.markdown('<div class="tech-news">', unsafe_allow_html=True)
-    st.markdown("### Latest in Tech & Computer Science")
-    for item in news:
-        st.markdown(f"- {item['title']}")
-    st.markdown('</div>', unsafe_allow_html=True)
+            except Exception as e:
+                st.error(f"An error occurred: {e}")
 
 st.markdown("""
-<div style='text-align: center; margin-top: 2rem; color: #8892b0;'>
-    Engineered with 🧠💡 by Your Superintelligent AI Code Assistant
+<div style='text-align: center; margin-top: 2rem; color: #4a5568;'>
+    Created with ❤️ by Your Sleek AI Code Assistant
 </div>
 """, unsafe_allow_html=True)