Update app.py

app.py

@@ -28,19 +28,16 @@ model = genai.GenerativeModel(
 )
 
 system_instruction = """
-You are an AI-powered terminal assistant. Your role is to interpret user requests and execute appropriate terminal commands. Follow these guidelines:
+You are an AI-powered terminal assistant. Your role is to interpret user requests and execute appropriate terminal commands or actions. 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
-   - show files: List files in the current directory
-   - hide files: Hide the file explorer
-   - new file: Create a new file
-   - edit file: Open a file for editing
+   - File creation and editing: For creating new files and adding content
 
-2. Provide the exact command to be executed, without any explanation or additional text.
+2. Provide the exact command to be executed or action 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.
 
@@ -50,17 +47,14 @@ You are an AI-powered terminal assistant. Your role is to interpret user request
 
 6. For Python script execution, provide the command to run the script (e.g., '!python script_name.py').
 
-7. For show files, respond with 'show files'.
+7. For file creation and editing requests:
+   - If asked to create a new file, respond with: "CREATE_FILE:filename.py"
+   - 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. For hide files, respond with 'hide files'.
+8. If a request is unclear or doesn't match any known command type, respond with "Unclear request. Please provide more details."
 
-9. For new file requests, respond with 'new file filename.ext', where filename.ext is the name provided by the user or a default name if not specified.
-
-10. For edit file requests, respond with 'edit filename.ext', where filename.ext is the name of the file to be edited.
-
-11. 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.
+Always respond with ONLY the command to be executed or action to be taken, nothing else.
 """
 
 chat = model.start_chat(history=[])
@@ -96,8 +90,20 @@ def execute_code():
             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"):
+            # 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 -m pip install"):
                 result = execute_command(ai_result[1:])  # Remove the leading '!'
             elif ai_result.startswith("git clone"):
                 result = execute_command(ai_result)
@@ -110,25 +116,6 @@ def execute_code():
                     result = f"Error: Directory not found: {new_dir}"
             elif ai_result.startswith("!python "):
                 result = execute_command(ai_result[1:])  # Remove the leading '!'
-            elif ai_result == "show files":
-                files = os.listdir(current_dir)
-                result = "Files in current directory:\n" + "\n".join(files)
-            elif ai_result == "hide files":
-                result = "Files hidden."
-            elif ai_result.startswith("new file "):
-                filename = ai_result[9:].strip()
-                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 "):
-                filename = ai_result[5:].strip()
-                filepath = os.path.join(current_dir, filename)
-                if os.path.exists(filepath):
-                    result = "Enter code:"
-                    return jsonify({"result": result, "action": "edit", "filename": filename})
-                else:
-                    result = f"Error: File {filename} not found."
             else:
                 result = f"Unclear AI response: {ai_result}"