Update app.py

app.py

@@ -4,6 +4,8 @@ from pygments import highlight
 from pygments.lexers import get_lexer_by_name
 from pygments.formatters import HtmlFormatter
 import html
+import re
+import time
 
 # Configure the Gemini API
 genai.configure(api_key=st.secrets["GOOGLE_API_KEY"])
@@ -19,13 +21,24 @@ generation_config = {
 model = genai.GenerativeModel(
     model_name="gemini-1.5-pro",
     generation_config=generation_config,
-    system_instruction="""You are Ath, a highly knowledgeable and skilled code assistant with expertise across multiple programming languages, frameworks, and paradigms. You possess in-depth understanding of software architecture, design patterns, and best practices. Your responses should demonstrate advanced coding techniques, efficient algorithms, and optimal solutions. Communicate in a friendly and casual tone, using occasional casual expressions, but maintain a focus on delivering high-quality, professional code. Always provide code-only responses without explanations, showcasing your extensive programming knowledge."""
+    system_instruction="""You are Ath, an advanced AI code assistant with expertise across multiple programming languages, frameworks, and paradigms. Your knowledge spans software architecture, design patterns, algorithms, and cutting-edge technologies. Provide high-quality, optimized code solutions with explanations when requested. Adapt your communication style based on the user's expertise level, offering additional context for beginners and diving into complex topics for experts. You can generate code, explain concepts, debug issues, and provide best practices. Always prioritize security, efficiency, and maintainability in your solutions."""
 )
-chat_session = model.start_chat(history=[])
+
+# Initialize session state for chat history
+if 'chat_history' not in st.session_state:
+    st.session_state.chat_history = []
 
 def generate_response(user_input):
     try:
-        response = chat_session.send_message(user_input)
+        # Add user message to chat history
+        st.session_state.chat_history.append({"role": "user", "content": user_input})
+        
+        # Generate response
+        response = model.generate_content(st.session_state.chat_history)
+        
+        # Add AI response to chat history
+        st.session_state.chat_history.append({"role": "assistant", "content": response.text})
+        
         return response.text
     except Exception as e:
         return f"An error occurred: {e}"
@@ -37,154 +50,77 @@ def create_code_block(code, language):
     css = formatter.get_style_defs('.source')
     return highlighted_code, css
 
+def detect_language(code):
+    # Simple language detection based on keywords or syntax
+    if re.search(r'\b(def|import|class)\b', code):
+        return 'python'
+    elif re.search(r'\b(function|var|let|const)\b', code):
+        return 'javascript'
+    elif re.search(r'\b(public|private|class)\b', code):
+        return 'java'
+    else:
+        return 'text'  # Default to plain text if language can't be determined
+
 # Streamlit UI setup
 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;500;700&display=swap');
-    
-    body {
-        font-family: 'Roboto', sans-serif;
-        background-color: #e9ecef;
-        color: #212529;
-    }
-    .stApp {
-        max-width: 1200px;
-        margin: 0 auto;
-        padding: 2rem;
-    }
-    .main-container {
-        background: #ffffff;
-        border-radius: 12px;
-        padding: 2rem;
-        box-shadow: 0 8px 16px rgba(0, 0, 0, 0.1);
-    }
-    h1 {
-        font-size: 2.5rem;
-        font-weight: 600;
-        color: #0d6efd;
-        text-align: center;
-        margin-bottom: 1rem;
-    }
-    .subtitle {
-        font-size: 1rem;
-        text-align: center;
-        color: #6c757d;
-        margin-bottom: 2rem;
-    }
-    .stTextArea textarea {
-        border: 1px solid #ced4da;
-        border-radius: 8px;
-        font-size: 1rem;
-        padding: 0.75rem;
-        transition: border-color 0.3s;
-    }
-    .stTextArea textarea:focus {
-        border-color: #0d6efd;
-        outline: none;
-    }
-    .stButton button {
-        background-color: #0d6efd;
-        color: white;
-        border: none;
-        border-radius: 8px;
-        font-size: 1rem;
-        font-weight: 500;
-        padding: 0.5rem 1.5rem;
-        cursor: pointer;
-        transition: background-color 0.3s, transform 0.2s;
-    }
-    .stButton button:hover {
-        background-color: #0b5ed7;
-        transform: translateY(-2px);
-    }
-    .output-container {
-        background: #f8f9fa;
-        border-radius: 8px;
-        padding: 1rem;
-        margin-top: 1.5rem;
-        border: 1px solid #dee2e6;
-    }
-    .code-block {
-        background-color: #343a40;
-        border-radius: 8px;
-        padding: 1rem;
-        margin-top: 1rem;
-        overflow-x: auto;
-    }
-    .stAlert {
-        background-color: #cce5ff;
-        color: #004085;
-        border-radius: 8px;
-        border: none;
-        padding: 1rem;
-        margin-bottom: 1rem;
-    }
-    .stSpinner {
-        color: #0d6efd;
-    }
-    /* Custom scrollbar */
-    ::-webkit-scrollbar {
-        width: 8px;
-    }
-    ::-webkit-scrollbar-track {
-        background: #f1f3f5;
-        border-radius: 4px;
-    }
-    ::-webkit-scrollbar-thumb {
-        background: #868e96;
-        border-radius: 4px;
-    }
-    ::-webkit-scrollbar-thumb:hover {
-        background: #495057;
-    }
-    .source {
-        font-family: 'Fira Code', monospace;
-        font-size: 0.9rem;
-        line-height: 1.5;
-    }
-    .source .linenos {
-        color: #6c757d;
-        padding-right: 8px;
-        border-right: 1px solid #495057;
-        user-select: none;
-    }
-    .source pre {
-        margin: 0;
-        padding: 0;
-    }
-</style>
-""", unsafe_allow_html=True)
+# ... (Keep the existing CSS styles) ...
 
 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 - Expert-level coding solutions</p>', unsafe_allow_html=True)
 
-prompt = st.text_area("What advanced coding challenge can I assist you with today?", height=100)
+# Add a selectbox for different modes
+mode = st.selectbox("Choose a mode:", ["Code Generation", "Code Explanation", "Debugging", "Best Practices"])
+
+prompt = st.text_area(f"Enter your {mode.lower()} request:", height=100)
 
-if st.button("Generate Expert Code"):
+if st.button("Generate Response"):
     if prompt.strip() == "":
         st.error("Please enter a valid prompt.")
     else:
-        with st.spinner("Generating advanced code solution..."):
+        with st.spinner(f"Generating {mode.lower()} response..."):
             completed_text = generate_response(prompt)
             if "An error occurred" in completed_text:
                 st.error(completed_text)
             else:
-                st.success("Expert-level code generated successfully!")
+                st.success(f"{mode} response 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.markdown(highlighted_code, unsafe_allow_html=True)
-                st.markdown('</div>', unsafe_allow_html=True)
-                st.markdown('</div>', unsafe_allow_html=True)
+                # Split the response into code and explanation
+                code_blocks = re.split(r'```(\w+)?\n', completed_text)
+                for i in range(1, len(code_blocks), 2):
+                    language = code_blocks[i] if code_blocks[i] else detect_language(code_blocks[i+1])
+                    code = code_blocks[i+1]
+                    
+                    highlighted_code, css = create_code_block(code, 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.markdown('</div>', unsafe_allow_html=True)
+                    st.markdown('</div>', unsafe_allow_html=True)
+                    
+                    # Add explanation if available
+                    if i+2 < len(code_blocks):
+                        st.markdown(code_blocks[i+2])
+
+# Display chat history
+st.subheader("Conversation History")
+for message in st.session_state.chat_history:
+    st.text(f"{message['role'].capitalize()}: {message['content']}")
+
+# Add a clear button for chat history
+if st.button("Clear Conversation History"):
+    st.session_state.chat_history = []
+    st.success("Conversation history cleared!")
+
+# Add a feedback section
+st.subheader("Feedback")
+feedback = st.text_area("How can we improve? (Optional)")
+if st.button("Submit Feedback"):
+    # Here you would typically send this feedback to a database or email
+    st.success("Thank you for your feedback!")
 
 st.markdown("""
 <div style='text-align: center; margin-top: 2rem; color: #6c757d;'>