Update app.py

app.py

@@ -6,10 +6,10 @@ genai.configure(api_key=st.secrets["GOOGLE_API_KEY"])
 
 # Create the model with system instructions
 generation_config = {
-    "temperature": 1,
+    "temperature": 0.7,
     "top_p": 0.95,
     "top_k": 64,
-    "max_output_tokens": 8192,
+    "max_output_tokens": 10240,
 }
 
 model = genai.GenerativeModel(
@@ -20,8 +20,11 @@ model = genai.GenerativeModel(
 chat_session = model.start_chat(history=[])
 
 def generate_response(user_input):
-    response = chat_session.send_message(user_input)
-    return response.text
+    try:
+        response = chat_session.send_message(user_input)
+        return response.text
+    except Exception as e:
+        return f"An error occurred: {e}"
 
 # Streamlit UI setup
 st.set_page_config(page_title="Sleek AI Code Assistant", page_icon="💻", layout="wide")
@@ -125,13 +128,16 @@ if st.button("Generate Code"):
     else:
         with st.spinner("Generating code..."):
             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 "An error occurred" in completed_text:
+                st.error(completed_text)
+            else:
+                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)
 
 st.markdown("""
 <div style='text-align: center; margin-top: 2rem; color: #4a5568;'>