Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import google.generativeai as genai
 import traceback
+import subprocess
 
 # Configure the Gemini API with advanced error handling
 try:
@@ -11,10 +12,10 @@ except Exception as e:
 
 # Create the model with system instructions and advanced configuration
 generation_config = {
-    "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
+    "temperature": 0.3,  # Lower temperature for more deterministic responses
+    "top_p": 0.85,
+    "top_k": 40,
+    "max_output_tokens": 12288,  # Increased max output tokens for longer responses
 }
 
 model = genai.GenerativeModel(
@@ -33,6 +34,16 @@ def generate_response(user_input):
         st.error(traceback.format_exc())
         return None
 
+def execute_code(code):
+    try:
+        result = subprocess.run(['python', '-c', code], capture_output=True, text=True, timeout=10)
+        if result.returncode == 0:
+            return result.stdout
+        else:
+            return result.stderr
+    except Exception as e:
+        return f"An error occurred while executing code: {e}"
+
 # Streamlit UI setup
 st.set_page_config(page_title="Sleek AI Code Assistant", page_icon="🚀", layout="wide")
 
@@ -160,6 +171,19 @@ for entry in st.session_state.history:
     st.markdown(f"**Assistant:**")
     st.code(entry['assistant'])
 
+# Interactive Code Execution
+if st.session_state.history:
+    st.markdown("## Execute Code")
+    code_to_execute = st.selectbox("Select code to execute", [entry['assistant'] for entry in st.session_state.history])
+    if st.button("Execute Code"):
+        with st.spinner("Executing code..."):
+            execution_result = execute_code(code_to_execute)
+            st.markdown('<div class="output-container">', unsafe_allow_html=True)
+            st.markdown('<div class="code-block">', unsafe_allow_html=True)
+            st.code(execution_result)
+            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;'>
     Created with ❤️ by Your Sleek AI Code Assistant