Update app.py

app.py

@@ -3,15 +3,8 @@ import subprocess
 import random
 import json
 from flask import Flask, render_template
-
-app = Flask(__name__)
-
-@app.route("/")
-def index():
-    return render_template("index.html")
-    
 from datetime import datetime
-from gradio import blocks  # Import blocks 
+from gradio import Blocks
 from safe_search import safe_search
 from i_search import google, i_search as i_s
 from agent import (
@@ -28,8 +21,16 @@ from agent import (
     UNDERSTAND_TEST_RESULTS_PROMPT,
 )
 from utils import parse_action, parse_file_content, read_python_module_structure
+from huggingface_hub import cached_download, hf_hub_url
+from transformers import InferenceClient
+
+app = Flask(__name__)
+
+@app.route("/")
+def index():
+    return render_template("index.html")
 
-class App(blocks.Blocks):  # Inherit from gradio.blocks.Blocks
+class App(Blocks):
     def __init__(self):
         super().__init__()
         self.app_state = {"components": []}
@@ -128,108 +129,14 @@ class App(blocks.Blocks):  # Inherit from gradio.blocks.Blocks
     def run_terminal_command(self, command, history):
         output = ""
         try:
-            if command.startswith("add "):
-                component_type = command.split("add ")[1]
-                return self.add_component(component_type)
-            elif command.startswith("search "):
-                query = command.split("search ")[1]
-                return google(query)
-            elif command.startswith("i search "):
-                query = command.split("i search ")[1]
-                return i_s(query)
-            elif command.startswith("safe search "):
-                query = command.split("safesearch ")[1]
-                return safe_search(query)
-            elif command.startswith("read "):
-                file_path = command.split("read ")[1]
-                return parse_file_content(file_path)
-            elif command == "task":
-                return TASK_PROMPT
-            elif command == "modify":
-                return MODIFY_PROMPT
-            elif command == "log":
-                return LOG_PROMPT
-            elif command.startswith("understand test results "):
-                test_results = command.split("understand test results ")[1]
-                return self.understand_test_results(test_results)
-            elif command.startswith("compress history"):
-                return self.compress_history(history)
-            elif command == "help":
-                return self.get_help_message()
-            elif command == "exit":
-                exit()
-            else:
-                output = subprocess.check_output(command, shell=True).decode("utf-8")
+            process = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+            output, error = process.communicate()
+            if error:
+                output = error
         except Exception as e:
             output = str(e)
-        return output or "No output\n"
-
-    def compress_history(self, history):
-        compressed_history = ""
-        lines = history.strip().split("\n")
-        for line in lines:
-            if not line.strip().startswith("#"):
-                compressed_history += line + "\n"
-        return compressed_history
-
-    def understand_test_results(self, test_results):
-        # Logic to understand test results
-        return UNDERSTAND_TEST_RESULTS_PROMPT
-
-    def get_help_message(self):
-        return """
-        Available commands:
-        - add [component_type]: Add a component to the app canvas
-        - search [query]: Perform a Google search
-        - i search [query]: Perform an intelligent search
-        - safe search [query]: Perform a safe search
-        - read [file_path]: Read and parse the content of a Python module
-        - task: Prompt for a task to perform
-        - modify: Prompt to modify a component property
-        - log: Prompt to log a response
-        - understand test results [test_results]: Understand test results
-        - compress history: Compress the terminal history by removing comments
-        - help: Show this help message
-        - exit: Exit the program
-        """
-
-    def process_input(self, input_text):
-        if input_text.strip().startswith("/"):
-            command = input_text.strip().lstrip("/")
-            output = self.run_terminal_command(command, self.terminal_history)
-            self.terminal_history += f"{input_text}\n{output}\n"
-            return output
-        else:
-            model_index = random.randint(0, len(self.nlp_models)-1)
-            response = self.get_nlp_response(input_text, model_index)
-            component_id, action, property_name, property_value = parse_action(response)
-            if component_id:
-                component = next((comp for comp in self.app_state["components"] if comp["id"] == component_id), None)
-                if component:
-                    if action == "update":
-                        component["properties"][property_name] = property_value
-                        return (
-                            self.update_app_canvas(),
-                            f"System: Updated property '{property_name}' of component with ID {component_id}\n",
-                        )
-                    elif action == "remove":
-                        self.app_state["components"].remove(component)
-                        return (
-                            self.update_app_canvas(),
-                            f"System: Removed component with ID {component_id}\n",
-                        )
-                    else:
-                        return None, f"Error: Invalid action: {action}\n"
-                else:
-                    return None, f"Error: Component with ID {component_id} not found\n"
-            else:
-                return None, f"Error: Failed to parse action from NLP response\n"
-
-def run(self):
-    print("Welcome to the Python App Builder!")
-    print("Type 'help' to see the available commands.")
-    print("-" * 50)
-    self.launch()
+        self.terminal_history += f"{command}\n{output.decode('utf-8')}\n"
+        return self.terminal_history
 
 if __name__ == "__main__":
     app.run(debug=True)