Spaces:

Athagi
/

Agi-v1

Sleeping

App Files Files Community

Athagi commited on Jan 29

Commit

5afdcff

verified ·

1 Parent(s): 646abb1

Update app.py

Browse files

Files changed (1) hide show

app.py +99 -81

app.py CHANGED Viewed

@@ -1,88 +1,106 @@
-<!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>

+from flask import Flask, request, jsonify, render_template, send_from_directory
+import os
+import subprocess
+import tempfile
+import google.generativeai as genai
+app = Flask(__name__)
+temp_dir = tempfile.mkdtemp()
+current_dir = temp_dir
+# Configure Google Generative AI
+genai.configure(api_key=os.environ["GEMINI_API_KEY"])
+# Create the model
+generation_config = {
+    "temperature": 1,
+    "top_p": 0.95,
+    "top_k": 64,
+    "max_output_tokens": 8192,
+    "response_mime_type": "text/plain",
+}
+model = genai.GenerativeModel(
+    model_name="gemini-1.5-pro-latest",
+    generation_config=generation_config,
+    system_instruction="You are an AGI assistant. Follow these rules:\n1. Respond with executable code when asked to create files\n2. Use Python code blocks for scripts\n3. Use bash code blocks for terminal commands\n4. Keep responses concise and actionable"
+)
+chat_session = model.start_chat(history=[])
+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")
+def handle_code_response(response):
+    python_code = []
+    for part in response.parts:
+        if "```python" in part.text:
+            code = part.text.split("```python")[1].split("```")[0].strip()
+            python_code.append(code)
+    return python_code
+@app.route("/execute", methods=["POST"])
+def execute_code():
+    command = request.json.get("code", "").strip()
+    if not command:
+        return jsonify({"result": "Error: No command provided."})
+    try:
+        if command.lower().startswith(("ai:", "/ai")):
+            prompt = command.split(":", 1)[1].strip()
+            response = chat_session.send_message(prompt)
+            python_code = handle_code_response(response)
+            if python_code:
+                filename = "generated_script.py"
+                with open(os.path.join(current_dir, filename), "w") as f:
+                    f.write("\n".join(python_code))
+                return jsonify({
+                    "result": response.text,
+                    "type": "code",
+                    "file": filename
+                })
+            return jsonify({
+                "result": response.text,
+                "type": "text"
+            })
+        elif command.lower().startswith(("file:", "/file")):
+            _, operation = command.split(":", 1)
+            if "list" in operation:
+                return jsonify({
+                    "result": "\n".join(os.listdir(current_dir)),
+                    "type": "file_list"
+                })
+            return jsonify({"result": "Unsupported file operation", "type": "error"})
+        elif command.startswith("!"):
+            result = execute_command(command[1:])
+            return jsonify({"result": result, "type": "command_output"})
+        else:
+            return jsonify({"result": "Unrecognized command. Type 'help' for options.", "type": "error"})
+    except Exception as e:
+        return jsonify({"result": f"Error: {str(e)}", "type": "error"})
+@app.route("/download/<path:filename>")
+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)