Spaces:

Athagi
/

Agi-v1

Sleeping

App Files Files Community

Athagi commited on Jan 29

Commit

ef280d5

verified ·

1 Parent(s): 06cfe95

Update app.py

Browse files

Files changed (1) hide show

app.py +87 -130

app.py CHANGED Viewed

@@ -1,131 +1,88 @@
-from flask import Flask, request, jsonify, render_template, send_from_directory
-import os
-import subprocess
-import tempfile
-import shutil
-import sys
-import torch
-import platform
-from huggingface_hub import InferenceClient
-import re
-app = Flask(__name__)
-temp_dir = tempfile.mkdtemp()
-current_dir = temp_dir
-# Initialize Hugging Face client
-client = InferenceClient(
-    token=os.environ.get("HF_API_KEY", "hf_xxxxxxxxxxxxxxxxxxxxxxxx")
-)
-# System State
-AGI_STATE = {
-    "active_processes": {},
-    "tts_engine_loaded": False,
-    "current_model": "google/gemma-2-2b-it"
-}
-def execute_command(command, cwd=None):
-    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:", "/ai")):
-            return handle_ai_command(command)
-        elif command.lower().startswith(("tts:", "/tts")):
-            return handle_tts_command(command)
-        elif command.lower().startswith(("sys:", "/sys")):
-            return handle_system_command(command)
-        elif command.lower().startswith(("web:", "/web")):
-            return handle_web_command(command)
-        elif command.lower().startswith(("file:", "/file")):
-            return handle_file_command(command)
-        else:
-            return jsonify({"result": execute_generic_command(command)})
-    except Exception as e:
-        return jsonify({"result": f"Error: {str(e)}"})
-def handle_ai_command(command):
-    prompt = command.split(":", 1)[1].strip()
-    response = client.text_generation(
-        model=AGI_STATE["current_model"],
-        prompt=f"{read_knowledge_base()}\n\nUser: {prompt}\nAssistant:",
-        max_new_tokens=500
-    )
-    return jsonify({
-        "result": response,
-        "type": "ai_response"
-    })
-def handle_tts_command(command):
-    return jsonify({"result": "TTS functionality not implemented", "type": "error"})
-def handle_system_command(command):
-    cmd = command.split(":", 1)[1].strip()
-    if cmd == "processes":
-        return jsonify({
-            "type": "process_list",
-            "processes": list(AGI_STATE["active_processes"].keys())
-        })
-    return jsonify({"result": "Unsupported system command", "type": "error"})
-def handle_web_command(command):
-    return jsonify({"result": "Web functionality not implemented", "type": "error"})
-def handle_file_command(command):
-    try:
-        _, operation, *rest = command.split(":", 2)
-        if operation == "list":
-            return jsonify({
-                "type": "file_list",
-                "files": os.listdir(current_dir)
-            })
-        return jsonify({"result": "Unsupported file operation", "type": "error"})
-    except Exception as e:
-        return jsonify({"result": f"File error: {str(e)}", "type": "error"})
-def execute_generic_command(command):
-    if command == "help":
-        return format_help_response()
-    elif command.startswith("!"):
-        return execute_command(command[1:])
-    return "Unrecognized command. Type 'help' for options."
-def format_help_response():
-    return """
-    Available Commands:
-    /ai [prompt] - Get AI response
-    /file:list - List files
-    !command - Execute shell command
-    """
-@app.route("/download/<path:filename>")
-def download_file(filename):
-    return send_from_directory(current_dir, filename, as_attachment=True)
-def read_knowledge_base():
-    with open('knowledge.txt') as f:
-        return f.read()
-if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860)

+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>AGI Terminal</title>
+    <link rel="stylesheet" href="{{ url_for('static', filename='style.css') }}">
+</head>
+<body>
+    <div class="terminal-container">
+        <div class="chat-output" id="chat">
+            <div class="message system">
+                🤖 AGI Terminal Ready<br>
+                Available commands:<br>
+                - AI: [your question]<br>
+                - file:list<br>
+                - ![command]
+            </div>
+        </div>
+        <div class="input-container">
+            <div class="prompt">>></div>
+            <input type="text" id="command-input" autofocus>
+            <button onclick="executeCommand()">Execute</button>
+        </div>
+    </div>
+    <script>
+        const chat = document.getElementById('chat');
+        const input = document.getElementById('command-input');
+        async function executeCommand() {
+            const command = input.value.trim();
+            if (!command) return;
+            addMessage(command, 'user');
+            input.value = '';
+            const loader = addLoader();
+            try {
+                const response = await fetch('/execute', {
+                    method: 'POST',
+                    headers: {'Content-Type': 'application/json'},
+                    body: JSON.stringify({ code: command })
+                });
+                const data = await response.json();
+                loader.remove();
+                handleResponse(data);
+            } catch (error) {
+                loader.remove();
+                addMessage(`Error: ${error.message}`, 'error');
+            }
+        }
+        function handleResponse(data) {
+            let content = data.result;
+            if (data.type === 'code') {
+                content += `<br><a href="/download/${data.file}" download class="download-link">Download ${data.file}</a>`;
+            }
+            addMessage(content, data.type);
+        }
+        function addMessage(content, type) {
+            const div = document.createElement('div');
+            div.className = `message ${type}`;
+            div.innerHTML = content;
+            chat.appendChild(div);
+            chat.scrollTop = chat.scrollHeight;
+        }
+        function addLoader() {
+            const loader = document.createElement('div');
+            loader.className = 'message loader';
+            loader.innerHTML = '<div class="dot-flashing"></div>';
+            chat.appendChild(loader);
+            return loader;
+        }
+        input.addEventListener('keypress', (e) => {
+            if (e.key === 'Enter') executeCommand();
+        });
+    </script>
+</body>
+</html>