Update app.py

app.py

@@ -2,6 +2,9 @@ import streamlit as st
 import google.generativeai as genai
 import traceback
 import subprocess
+import re
+import ast
+import astor
 
 # Configure the Gemini API with advanced error handling
 try:
@@ -12,10 +15,10 @@ except Exception as e:
 
 # Create the model with system instructions and advanced configuration
 generation_config = {
-    "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
+    "temperature": 0.1,  # Lower temperature for more deterministic responses
+    "top_p": 0.75,
+    "top_k": 20,
+    "max_output_tokens": 20480,  # Increased max output tokens for longer responses
 }
 
 model = genai.GenerativeModel(
@@ -44,6 +47,60 @@ def execute_code(code):
     except Exception as e:
         return f"An error occurred while executing code: {e}"
 
+def optimize_code(code):
+    try:
+        optimization_prompt = f"Optimize the following Python code:\n```python\n{code}\n```"
+        optimized_code = generate_response(optimization_prompt)
+        return optimized_code
+    except Exception as e:
+        st.error(f"An error occurred while optimizing code: {e}")
+        return None
+
+def debug_code(code):
+    try:
+        debug_prompt = f"Debug the following Python code and suggest fixes:\n```python\n{code}\n```"
+        debug_suggestions = generate_response(debug_prompt)
+        return debug_suggestions
+    except Exception as e:
+        st.error(f"An error occurred while debugging code: {e}")
+        return None
+
+def suggest_improvements(code):
+    try:
+        improvement_prompt = f"Suggest improvements for the following Python code:\n```python\n{code}\n```"
+        improvement_suggestions = generate_response(improvement_prompt)
+        return improvement_suggestions
+    except Exception as e:
+        st.error(f"An error occurred while suggesting improvements: {e}")
+        return None
+
+def analyze_code(code):
+    try:
+        tree = ast.parse(code)
+        analyzer = CodeAnalyzer()
+        analyzer.visit(tree)
+        return analyzer.report()
+    except Exception as e:
+        st.error(f"An error occurred while analyzing code: {e}")
+        return None
+
+class CodeAnalyzer(ast.NodeVisitor):
+    def __init__(self):
+        self.issues = []
+
+    def visit_FunctionDef(self, node):
+        if len(node.body) == 0:
+            self.issues.append(f"Function '{node.name}' is empty.")
+        self.generic_visit(node)
+
+    def visit_For(self, node):
+        if isinstance(node.target, ast.Name) and node.target.id == '_':
+            self.issues.append(f"Possible unused variable in loop at line {node.lineno}.")
+        self.generic_visit(node)
+
+    def report(self):
+        return "\n".join(self.issues)
+
 # Streamlit UI setup
 st.set_page_config(page_title="Sleek AI Code Assistant", page_icon="🚀", layout="wide")
 
@@ -184,6 +241,62 @@ if st.session_state.history:
             st.markdown('</div>', unsafe_allow_html=True)
             st.markdown('</div>', unsafe_allow_html=True)
 
+# Code Optimization
+if st.session_state.history:
+    st.markdown("## Optimize Code")
+    code_to_optimize = st.selectbox("Select code to optimize", [entry['assistant'] for entry in st.session_state.history])
+    if st.button("Optimize Code"):
+        with st.spinner("Optimizing code..."):
+            optimized_code = optimize_code(code_to_optimize)
+            if optimized_code:
+                st.markdown('<div class="output-container">', unsafe_allow_html=True)
+                st.markdown('<div class="code-block">', unsafe_allow_html=True)
+                st.code(optimized_code)
+                st.markdown('</div>', unsafe_allow_html=True)
+                st.markdown('</div>', unsafe_allow_html=True)
+
+# Code Debugging
+if st.session_state.history:
+    st.markdown("## Debug Code")
+    code_to_debug = st.selectbox("Select code to debug", [entry['assistant'] for entry in st.session_state.history])
+    if st.button("Debug Code"):
+        with st.spinner("Debugging code..."):
+            debug_suggestions = debug_code(code_to_debug)
+            if debug_suggestions:
+                st.markdown('<div class="output-container">', unsafe_allow_html=True)
+                st.markdown('<div class="code-block">', unsafe_allow_html=True)
+                st.code(debug_suggestions)
+                st.markdown('</div>', unsafe_allow_html=True)
+                st.markdown('</div>', unsafe_allow_html=True)
+
+# Code Improvement Suggestions
+if st.session_state.history:
+    st.markdown("## Suggest Improvements")
+    code_to_improve = st.selectbox("Select code to improve", [entry['assistant'] for entry in st.session_state.history])
+    if st.button("Suggest Improvements"):
+        with st.spinner("Suggesting improvements..."):
+            improvement_suggestions = suggest_improvements(code_to_improve)
+            if improvement_suggestions:
+                st.markdown('<div class="output-container">', unsafe_allow_html=True)
+                st.markdown('<div class="code-block">', unsafe_allow_html=True)
+                st.code(improvement_suggestions)
+                st.markdown('</div>', unsafe_allow_html=True)
+                st.markdown('</div>', unsafe_allow_html=True)
+
+# Code Analysis
+if st.session_state.history:
+    st.markdown("## Analyze Code")
+    code_to_analyze = st.selectbox("Select code to analyze", [entry['assistant'] for entry in st.session_state.history])
+    if st.button("Analyze Code"):
+        with st.spinner("Analyzing code..."):
+            analysis_result = analyze_code(code_to_analyze)
+            if analysis_result:
+                st.markdown('<div class="output-container">', unsafe_allow_html=True)
+                st.markdown('<div class="code-block">', unsafe_allow_html=True)
+                st.code(analysis_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