Update app.py

app.py

@@ -6,10 +6,6 @@ import shutil
 import sys
 import google.generativeai as genai
 
-def sanitize_command(command):
-    """Sanitize the command to prevent shell injection."""
-    return command.replace(";", "").replace("&&", "").replace("||", "")
-
 app = Flask(__name__)
 
 # Create a temporary directory for operations
@@ -41,7 +37,7 @@ You are an AI-powered terminal assistant. Your role is to interpret user request
    - Python script execution: For running .py files
    - File creation and editing: For creating new files and adding content
 
-2. Provide the exact command to be executed or action to be taken, without any explanation or additional text.
+2. Provide the exact command(s) to be executed or action(s) to be taken, without any explanation or additional text.
 
 3. For pip install requests, always prefix the command with '!python -m' to ensure proper execution.
 
@@ -56,18 +52,19 @@ You are an AI-powered terminal assistant. Your role is to interpret user request
    - If asked to edit a file with specific content, respond with: "EDIT_FILE:filename.py:file_content"
    Replace 'filename.py' with the actual filename and 'file_content' with the requested content.
 
-8. If a request is unclear or doesn't match any known command type, respond with "Unclear request. Please provide more details."
+8. If a request requires multiple commands, provide them as a list of commands, each on a new line, prefixed with 'CMD:'.
+
+9. 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 or action to be taken, nothing else.
+Always respond with ONLY the command(s) to be executed or action(s) to be taken, nothing else.
 """
 
 chat = model.start_chat(history=[])
 
 def execute_command(command, cwd=None):
     """Executes a command and returns the output."""
-    sanitized_command = sanitize_command(command)
     process = subprocess.Popen(
-        sanitized_command,
+        command,
         shell=True,
         stdout=subprocess.PIPE,
         stderr=subprocess.PIPE,
@@ -101,30 +98,39 @@ def execute_code():
             response = chat.send_message(f"{system_instruction}\n\nKnowledge Base:\n{knowledge_base}\n\nUser request: {ai_command}")
             ai_result = response.text.strip()
 
-            # Execute the AI-suggested command or action
-            if ai_result.startswith("CREATE_FILE:"):
-                filename = ai_result.split(":")[1]
-                filepath = os.path.join(current_dir, filename)
-                with open(filepath, 'w') as f:
-                    pass  # Create an empty file
-                result = f"Created new file: {filename}"
-            elif ai_result.startswith("EDIT_FILE:"):
-                _, filename, content = ai_result.split(":", 2)
-                filepath = os.path.join(current_dir, filename)
-                with open(filepath, 'w') as f:
-                    f.write(content)
-                result = f"File {filename} created and edited successfully."
-            elif ai_result.startswith("!python"):
-                result = execute_command(ai_result[1:])  # Remove the leading '!'
-            elif ai_result.startswith("git clone"):
-                result = execute_command(ai_result)
-            elif ai_result.startswith("show files"):
-                files = os.listdir(current_dir)
-                result = "Files in current directory:\n" + "\n".join(files)
-            else:
-                result = f"Unclear AI response: {ai_result}"
-
-            return jsonify({"result": f"AI Executed: {ai_result}\n\nOutput:\n{result}"})
+            # Split multiple commands
+            commands = [cmd.strip() for cmd in ai_result.split('CMD:') if cmd.strip()]
+            results = []
+
+            for cmd in commands:
+                if cmd.startswith("CREATE_FILE:"):
+                    filename = cmd.split(":")[1]
+                    filepath = os.path.join(current_dir, filename)
+                    with open(filepath, 'w') as f:
+                        pass  # Create an empty file
+                    results.append(f"Created new file: {filename}")
+                elif cmd.startswith("EDIT_FILE:"):
+                    _, filename, content = cmd.split(":", 2)
+                    filepath = os.path.join(current_dir, filename)
+                    with open(filepath, 'w') as f:
+                        f.write(content)
+                    results.append(f"File {filename} created and edited successfully.")
+                elif cmd.startswith("!"):
+                    results.append(execute_command(cmd[1:]))  # Remove the leading '!'
+                elif cmd.startswith("show files"):
+                    files = os.listdir(current_dir)
+                    results.append("Files in current directory:\n" + "\n".join(files))
+                elif cmd.startswith("cd "):
+                    new_dir = os.path.join(current_dir, cmd[3:])
+                    if os.path.isdir(new_dir):
+                        current_dir = os.path.abspath(new_dir)
+                        results.append(f"Changed directory to: {current_dir}")
+                    else:
+                        results.append(f"Error: Directory not found: {new_dir}")
+                else:
+                    results.append(execute_command(cmd))
+
+            return jsonify({"result": f"AI Executed:\n{ai_result}\n\nOutput:\n" + "\n".join(results)})
         elif command == "show files":
             files = os.listdir(current_dir)
             return jsonify({"result": "Files in current directory:\n" + "\n".join(files)})