Update app.py

app.py

@@ -1,19 +1,20 @@
 import streamlit as st
 import google.generativeai as genai
+import subprocess
 
 # Configure the Gemini API
 genai.configure(api_key=st.secrets["GOOGLE_API_KEY"])
 
 # Create the model with system instructions
 generation_config = {
-    "temperature": 0.7,
+    "temperature": 0.5,
     "top_p": 0.95,
     "top_k": 64,
-    "max_output_tokens": 10240,
+    "max_output_tokens": 12288,
 }
 
 model = genai.GenerativeModel(
-    model_name="gemini-1.5-pro",
+    model_name="gemini-2.0-pro",  # Assuming a more advanced model is available
     generation_config=generation_config,
     system_instruction="You are a human named Ath who is 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 code only, without explanations.",
 )
@@ -26,6 +27,17 @@ def generate_response(user_input):
     except Exception as e:
         return f"An error occurred: {e}"
 
+def validate_code(code, language):
+    try:
+        if language == "Python":
+            result = subprocess.run(['python', '-c', code], capture_output=True, text=True)
+        elif language == "JavaScript":
+            result = subprocess.run(['node', '-e', code], capture_output=True, text=True)
+        # Add more languages as needed
+        return result.stdout, result.stderr
+    except Exception as e:
+        return None, f"Validation error: {e}"
+
 # Streamlit UI setup
 st.set_page_config(page_title="Sleek AI Code Assistant", page_icon="💻", layout="wide")
 
@@ -121,6 +133,7 @@ st.title("💻 Sleek 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)
+language = st.selectbox("Select programming language", ["Python", "JavaScript", "Other"])
 
 if st.button("Generate Code"):
     if prompt.strip() == "":
@@ -138,6 +151,14 @@ if st.button("Generate Code"):
                 st.code(completed_text)
                 st.markdown('</div>', unsafe_allow_html=True)
                 st.markdown('</div>', unsafe_allow_html=True)
+                
+                if language in ["Python", "JavaScript"]:
+                    with st.spinner("Validating code..."):
+                        stdout, stderr = validate_code(completed_text, language)
+                        if stdout:
+                            st.info(f"Validation Output:\n{stdout}")
+                        if stderr:
+                            st.warning(f"Validation Errors:\n{stderr}")
 
 st.markdown("""
 <div style='text-align: center; margin-top: 2rem; color: #4a5568;'>