from flask import Flask, request, jsonify, render_template, send_from_directory
import os
import subprocess
import tempfile
import shutil
import sys
import google.generativeai as genai

app = Flask(__name__)

# Create a temporary directory for operations
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(
        command,
        shell=True,
        stdout=subprocess.PIPE,
        stderr=subprocess.PIPE,
        text=True,
        cwd=cwd or current_dir
    )
    stdout, stderr = process.communicate()
    return stdout + stderr

@app.route("/")
def index():
    return render_template("index.html")

@app.route("/execute", methods=["POST"])
def execute_code():
    global current_dir
    command = request.json.get("code", "").strip()
    if not command:
        return jsonify({"result": "Error: No command provided."})

    try:
        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":
            return jsonify({"result": "Files hidden."})
        elif command.startswith("new file "):
            filename = command[9:].strip()
            filepath = os.path.join(current_dir, filename)
            with open(filepath, 'w') as f:
                pass  # Create an empty file
            return jsonify({"result": f"Created new file: {filename}"})
        elif command.startswith("edit "):
            filename = command[5:].strip()
            filepath = os.path.join(current_dir, filename)
            if os.path.exists(filepath):
                return jsonify({"result": "Enter code:", "action": "edit", "filename": filename})
            else:
                return jsonify({"result": f"Error: File {filename} not found."})
        elif command.startswith("cd "):
            new_dir = os.path.join(current_dir, command[3:])
            if os.path.isdir(new_dir):
                current_dir = os.path.abspath(new_dir)
                return jsonify({"result": f"Changed directory to: {current_dir}"})
            else:
                return jsonify({"result": f"Error: Directory not found: {new_dir}"})
        elif command.startswith("!"):
            result = execute_command(command[1:])
        elif command.startswith("pip install"):
            result = execute_command(f"{sys.executable} -m {command}")
        elif command.startswith("git "):
            result = execute_command(command)
        else:
            if command.endswith(".py"):
                result = execute_command(f"{sys.executable} {command}")
            else:
                result = execute_command(f"{sys.executable} -c \"{command}\"")
        return jsonify({"result": result})
    except Exception as e:
        return jsonify({"result": f"Error: {str(e)}"})

@app.route("/save_file", methods=["POST"])
def save_file():
    filename = request.json.get("filename")
    content = request.json.get("content")
    filepath = os.path.join(current_dir, filename)
    with open(filepath, 'w') as f:
        f.write(content)
    return jsonify({"result": f"File {filename} saved successfully."})

@app.route("/cleanup", methods=["POST"])
def cleanup():
    global temp_dir, current_dir
    if os.path.exists(temp_dir):
        shutil.rmtree(temp_dir)
    temp_dir = tempfile.mkdtemp()
    current_dir = temp_dir
    return jsonify({"result": "Temporary files cleaned up."})

@app.route("/list_files", methods=["GET"])
def list_files():
    files = os.listdir(current_dir)
    return jsonify({"files": files})

@app.route("/download/<path:filename>", methods=["GET"])
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)