Update app.py

app.py

@@ -4,6 +4,7 @@ import subprocess
 import tempfile
 import shutil
 import sys
+import google.generativeai as genai
 
 app = Flask(__name__)
 
@@ -11,6 +12,47 @@ app = Flask(__name__)
 temp_dir = tempfile.mkdtemp()
 current_dir = temp_dir
 
+# Configure Gemini AI
+genai.configure(api_key=os.environ["GEMINI_API_KEY"])
+
+generation_config = {
+    "temperature": 0.9,
+    "top_p": 1,
+    "top_k": 40,
+    "max_output_tokens": 1024,
+}
+
+model = genai.GenerativeModel(
+    model_name="gemini-1.5-pro",
+    generation_config=generation_config,
+)
+
+system_instruction = """
+You are an AI-powered terminal assistant. Your role is to interpret user requests and execute appropriate terminal commands. Follow these guidelines:
+
+1. Understand and execute various commands, including but not limited to:
+   - pip install: For installing Python packages
+   - git clone: For cloning repositories
+   - cd: For changing directories
+   - Python script execution: For running .py files
+
+2. Provide the exact command to be executed, without any explanation or additional text.
+
+3. For pip install requests, always prefix the command with '!python -m' to ensure proper execution.
+
+4. For git clone requests, provide the full 'git clone' command with the repository URL.
+
+5. For cd requests, simply provide the 'cd' command with the specified directory.
+
+6. For Python script execution, provide the command to run the script (e.g., '!python script_name.py').
+
+7. If a request is unclear or doesn't match any known command type, respond with "Unclear request. Please provide more details."
+
+Always respond with ONLY the command to be executed, nothing else.
+"""
+
+chat = model.start_chat(history=[])
+
 def execute_command(command, cwd=None):
     """Executes a command and returns the output."""
     process = subprocess.Popen(
@@ -36,7 +78,31 @@ def execute_code():
         return jsonify({"result": "Error: No command provided."})
 
     try:
-        if command == "show files":
+        if command.lower().startswith("ai:"):
+            # Process command with Gemini AI
+            ai_command = command[3:].strip()
+            response = chat.send_message(system_instruction + f"\nUser request: {ai_command}")
+            ai_result = response.text.strip()
+            
+            # Execute the AI-suggested command
+            if ai_result.startswith("!python -m pip install"):
+                result = execute_command(ai_result[1:])  # Remove the leading '!'
+            elif ai_result.startswith("git clone"):
+                result = execute_command(ai_result)
+            elif ai_result.startswith("cd "):
+                new_dir = os.path.join(current_dir, ai_result[3:])
+                if os.path.isdir(new_dir):
+                    current_dir = os.path.abspath(new_dir)
+                    result = f"Changed directory to: {current_dir}"
+                else:
+                    result = f"Error: Directory not found: {new_dir}"
+            elif ai_result.startswith("!python "):
+                result = execute_command(ai_result[1:])  # Remove the leading '!'
+            else:
+                result = f"Unclear AI response: {ai_result}"
+            
+            return jsonify({"result": f"AI Executed: {ai_result}\n\nOutput:\n{result}"})
+        elif command == "show files":
             files = os.listdir(current_dir)
             return jsonify({"result": "Files in current directory:\n" + "\n".join(files)})
         elif command == "hide files":
@@ -104,4 +170,5 @@ def download_file(filename):
     return send_from_directory(current_dir, filename, as_attachment=True)
 
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860)
+    app.run(host="0.0.0.0", port=7860)
+