Update app.py

app.py

@@ -1,18 +1,69 @@
-import os
 import subprocess
-import random
+from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
+import black
+from pylint import lint
+from io import StringIO
+import os
 import json
-from datetime import datetime
-from huggingface_hub import InferenceClient, cached_download, hf_hub_url
-import gradio as gr
-from safe_search import safe_search
-from i_search import google, i_search as i_s
+from streamlit_ace import st_ace
 from agent import (
-    ACTION_PROMPT, ADD_PROMPT, COMPRESS_HISTORY_PROMPT, LOG_PROMPT,
-    LOG_RESPONSE, MODIFY_PROMPT, PRE_PREFIX, SEARCH_QUERY, READ_PROMPT,
-    TASK_PROMPT, UNDERSTAND_TEST_RESULTS_PROMPT
+    AppType,
+    createLlamaPrompt,
+    createSpace,
+    isPythonOrGradioAppPrompt,
+    isReactAppPrompt,
+    isStreamlitAppPrompt,
+    generateFiles,
 )
-from utils import parse_action, parse_file_content, read_python_module_structure
+
+import importlib
+import sys
+
+def initialize_global_variables():
+    global HUGGING_FACE_REPO_URL, PROJECT_ROOT, AGENT_DIRECTORY
+    HUGGING_FACE_REPO_URL = "https://huggingface.co/spaces/acecalisto3/Mistri"
+    PROJECT_ROOT = "projects"
+    AGENT_DIRECTORY = "agents"
+
+initialize_global_variables()
+
+# Initialize session state attributes
+for attr in ['chat_history', 'terminal_history', 'workspace_projects', 'available_agents', 'current_state']:
+    if attr not in st.session_state:
+        st.session_state[attr] = []
+
+def save_agent_to_file(agent):
+    agents_path = os.path.join(PROJECT_ROOT, AGENT_DIRECTORY)
+    if not os.path.exists(agents_path):
+        os.makedirs(agents_path)
+
+    agent_file = os.path.join(agents_path, f"{agent.name}.txt")
+    config_file = os.path.join(agents_path, f"{agent.name}Config.txt")
+
+    with open(agent_file, "w") as file:
+        file.write(agent.create_agent_prompt())
+
+    with open(config_file, "w") as file:
+        file.write(f"Agent Name: {agent.name}\nDescription: {agent.description}")
+
+    st.session_state.available_agents.append(agent.name)
+
+    commit_and_push_changes(f"Add agent {agent.name}")
+
+def load_agent_prompt(agent_name):
+    agent_file = os.path.join(AGENT_DIRECTORY, f"{agent_name}.txt")
+    if os.path.exists(agent_file):
+        with open(agent_file, "r") as file:
+            agent_prompt = file.read()
+        return agent_prompt
+    else:
+        return None
+
+def create_agent_from_text(name, text):
+    skills = text.split('\n')
+    agent = AIAgent(name, "AI agent created from text input.", skills)
+    save_agent_to_file(agent)
+    return agent.create_agent_prompt()
 
 # Global Variables
 terminal_history = ""