app.py

import os
import gradio as gr  
from dotenv import load_dotenv
from openai import OpenAI
import shutil
import subprocess

load_dotenv()


client = OpenAI()

SYSTEM_PROMPT = """
You are a helpful assistant that converts natural language instructions into terminal commands.
Return ONLY the command itself without any explanations, backticks, or formatting.
Target System: Linux/Unix (Bash).

For dangerous commands (rm -rf, del, shutdown, format, dd, mkfs, etc.), prefix the command with ⚠️ warning icon.
Example: ⚠️ rm -rf /tmp/*

If the user asks a question that can be answered with a terminal command, provide the command.
Only reject requests that cannot be solved with terminal commands (jokes, stories, personal advice, etc.).
If the request cannot be solved with a terminal command, respond with: "Error: Not a valid terminal command request"
"""

def is_valid_command(command):
    # Remove warning icon if present
    clean_cmd = command.replace("⚠️", "").strip()
    
    # Get the base command (first word)
    base_cmd = clean_cmd.split()[0] if clean_cmd else ""
    
    # Check if command exists in system
    return shutil.which(base_cmd) is not None or base_cmd in ['cd', 'echo', 'export', 'alias', 'source']

def run_command_in_docker(command):
    # Remove warning icon
    clean_cmd = command.replace("⚠️", "").strip()
    
    try:
        result = subprocess.run(
            ["docker", "run", "--rm", "ubuntu:latest", "bash", "-c", clean_cmd],
            capture_output=True,
            text=True,
            timeout=10
        )
        
        if result.returncode == 0:
            return "✓ Command is valid"
        else:
            return f"Error: Command failed in Docker"
    except FileNotFoundError:
        return "⚠️ Docker not available - command not tested"
    except subprocess.TimeoutExpired:
        return "Error: Command timeout"
    except Exception as e:
        return f"Error: {str(e)}"

def generate_command(user_input):
    try:
        response = client.chat.completions.create(
            model="gpt-4o",
            messages=[
                {"role": "system", "content": SYSTEM_PROMPT},
                {"role": "user", "content": user_input}
            ],
            temperature=0
        )
        command = response.choices[0].message.content.strip()
        
        # Validate command
        if not is_valid_command(command):
            return f"Error: Invalid or unknown command generated"
        
        # Test command in Docker
        validation_result = run_command_in_docker(command)
        
        return f"{command}\n\n{validation_result}"
    except Exception as e:
        return f"Error: {str(e)}"

# ממשק Gradio
demo = gr.Interface(
    fn=generate_command,
    inputs=gr.Textbox(label="מה תרצה לבצע?"),
    outputs=gr.Code(label="פקודת CLI"),
    title="Natural Language to CLI Agent"
)

if __name__ == "__main__":
    demo.launch()