Update app.py

app.py

@@ -1,19 +1,20 @@
 import streamlit as st
 import google.generativeai as genai
+import traceback
 
-# Configure the Gemini API with error handling
+# Configure the Gemini API with advanced error handling
 try:
     genai.configure(api_key=st.secrets["GOOGLE_API_KEY"])
 except Exception as e:
     st.error(f"Failed to configure API: {e}")
     st.stop()
 
-# Create the model with system instructions
+# Create the model with system instructions and advanced configuration
 generation_config = {
-    "temperature": 0.7,  # Slightly lower temperature for more focused responses
-    "top_p": 0.95,
-    "top_k": 64,
-    "max_output_tokens": 8192,
+    "temperature": 0.5,  # Lower temperature for more deterministic responses
+    "top_p": 0.9,
+    "top_k": 50,
+    "max_output_tokens": 10240,  # Increased max output tokens for longer responses
 }
 
 model = genai.GenerativeModel(
@@ -28,7 +29,9 @@ def generate_response(user_input):
         response = chat_session.send_message(user_input)
         return response.text
     except Exception as e:
-        return f"An error occurred while generating response: {e}"
+        st.error(f"An error occurred while generating response: {e}")
+        st.error(traceback.format_exc())
+        return None
 
 # Streamlit UI setup
 st.set_page_config(page_title="Sleek AI Code Assistant", page_icon="🚀", layout="wide")
@@ -124,6 +127,10 @@ st.markdown('<div class="main-container">', unsafe_allow_html=True)
 st.title("🚀 Sleek AI Code Assistant")
 st.markdown('<p class="subtitle">Powered by Google Gemini</p>', unsafe_allow_html=True)
 
+# Session state to maintain conversation history
+if 'history' not in st.session_state:
+    st.session_state.history = []
+
 prompt = st.text_area("What code can I help you with today?", height=120)
 
 if st.button("Generate Code"):
@@ -133,15 +140,25 @@ if st.button("Generate Code"):
         with st.spinner("Generating code..."):
             try:
                 completed_text = generate_response(prompt)
-                st.success("Code generated successfully!")
-                
-                st.markdown('<div class="output-container">', unsafe_allow_html=True)
-                st.markdown('<div class="code-block">', unsafe_allow_html=True)
-                st.code(completed_text)
-                st.markdown('</div>', unsafe_allow_html=True)
-                st.markdown('</div>', unsafe_allow_html=True)
+                if completed_text:
+                    st.success("Code generated successfully!")
+                    st.session_state.history.append({"user": prompt, "assistant": completed_text})
+                    
+                    st.markdown('<div class="output-container">', unsafe_allow_html=True)
+                    st.markdown('<div class="code-block">', unsafe_allow_html=True)
+                    st.code(completed_text)
+                    st.markdown('</div>', unsafe_allow_html=True)
+                    st.markdown('</div>', unsafe_allow_html=True)
             except Exception as e:
                 st.error(f"An error occurred: {e}")
+                st.error(traceback.format_exc())
+
+# Display conversation history
+st.markdown("## Conversation History")
+for entry in st.session_state.history:
+    st.markdown(f"**User:** {entry['user']}")
+    st.markdown(f"**Assistant:**")
+    st.code(entry['assistant'])
 
 st.markdown("""
 <div style='text-align: center; margin-top: 2rem; color: #4a5568;'>