Update app.py

app.py

@@ -12,29 +12,30 @@ import spacy
 from spacy.lang.en import English
 import boto3
 import unittest
-from sympy import symbols, Eq, solve
+import docker
+import sympy as sp
 from scipy.optimize import minimize
 import numpy as np
 import matplotlib.pyplot as plt
 import seaborn as sns
+from IPython.display import display
 
 # Configure the Gemini API
 genai.configure(api_key=st.secrets["GOOGLE_API_KEY"])
 
 # Create the model with optimized parameters and enhanced system instructions
 generation_config = {
-    "temperature": 0.6,  # Lower temperature for more deterministic responses
-    "top_p": 0.8,        # Adjusted for better diversity
-    "top_k": 30,         # Increased for more diverse tokens
-    "max_output_tokens": 16384,  # Increased for longer responses
+    "temperature": 0.5,  # Lower temperature for more deterministic responses
+    "top_p": 0.7,        # Adjusted for better diversity
+    "top_k": 40,         # Increased for more diverse tokens
+    "max_output_tokens": 2048,  # Increased for longer responses
 }
 
 model = genai.GenerativeModel(
-    model_name="gemini-1.5-pro",
+    model_name="gemini-2.0-pro",
     generation_config=generation_config,
     system_instruction="""
-    You are Ath, 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 optimized, secure, and high-quality code only, without explanations. You are designed to provide accurate, efficient, and cutting-edge code solutions.
+    You are Ath, a highly knowledgeable and advanced code assistant. Your responses are optimized for secure, high-quality, and cutting-edge code solutions.
     """
 )
 chat_session = model.start_chat(history=[])
@@ -111,7 +112,6 @@ def interact_with_cloud_services(service_name, action, params):
 
 def run_tests():
     """Run automated tests using unittest."""
-    # Ensure the tests directory is importable
     tests_dir = os.path.join(os.getcwd(), 'tests')
     if not os.path.exists(tests_dir):
         os.makedirs(tests_dir)
@@ -125,11 +125,27 @@ def run_tests():
     test_result = test_runner.run(test_suite)
     return test_result
 
+def execute_code_in_docker(code):
+    """Execute code in a Docker container for safety and isolation."""
+    client = docker.from_env()
+    try:
+        container = client.containers.run(
+            image="python:3.9",
+            command=f"python -c '{code}'",
+            detach=True,
+            remove=True
+        )
+        result = container.wait()
+        logs = container.logs().decode('utf-8')
+        return logs, result['StatusCode']
+    except Exception as e:
+        return f"Error: {e}", 1
+
 def solve_equation(equation):
     """Solve mathematical equations using SymPy."""
-    x, y = symbols('x y')
-    eq = Eq(eval(equation))
-    solution = solve(eq, x)
+    x, y = sp.symbols('x y')
+    eq = sp.Eq(eval(equation))
+    solution = sp.solve(eq, x)
     return solution
 
 def optimize_function(function, initial_guess):
@@ -145,8 +161,19 @@ def visualize_data(data):
     plt.title('Correlation Heatmap')
     plt.show()
 
+def analyze_data(data):
+    """Perform advanced data analysis using Pandas and NumPy."""
+    df = pd.DataFrame(data)
+    summary = df.describe()
+    return summary
+
+def display_dataframe(data):
+    """Display a DataFrame in a user-friendly format."""
+    df = pd.DataFrame(data)
+    display(df)
+
 # Streamlit UI setup
-st.set_page_config(page_title="Sleek AI Code Assistant", page_icon="💻", layout="wide")
+st.set_page_config(page_title="Ultra AI Code Assistant", page_icon="🚀", layout="wide")
 
 st.markdown("""
 <style>
@@ -158,7 +185,7 @@ st.markdown("""
         color: #1a202c;
     }
     .stApp {
-        max-width: 1000px;
+        max-width: 1200px;
         margin: 0 auto;
         padding: 2rem;
     }
@@ -236,7 +263,7 @@ st.markdown("""
 """, unsafe_allow_html=True)
 
 st.markdown('<div class="main-container">', unsafe_allow_html=True)
-st.title("💻 Sleek AI Code Assistant")
+st.title("🚀 Ultra AI Code Assistant")
 st.markdown('<p class="subtitle">Powered by Google Gemini</p>', unsafe_allow_html=True)
 
 prompt = st.text_area("What code can I help you with today?", height=120)
@@ -271,12 +298,20 @@ if st.button("Generate Code"):
                         st.success("All tests passed successfully!")
                     else:
                         st.error("Some tests failed. Please check the code.")
+                    
+                    # Execute code in Docker
+                    execution_result, status_code = execute_code_in_docker(optimized_code)
+                    if status_code == 0:
+                        st.success("Code executed successfully in Docker!")
+                        st.text(execution_result)
+                    else:
+                        st.error(f"Code execution failed: {execution_result}")
             except Exception as e:
                 handle_error(e)
 
 st.markdown("""
 <div style='text-align: center; margin-top: 2rem; color: #4a5568;'>
-    Created with ❤️ by Your Sleek AI Code Assistant
+    Created with ❤️ by Your Ultra AI Code Assistant
 </div>
 """, unsafe_allow_html=True)