Update app.py

app.py

@@ -3,13 +3,9 @@ import subprocess
 import random
 import json
 from datetime import datetime
-
-from huggingface_hub import InferenceClient, cached_download, hf_hub_url
-import gradio as gr
-
+from gradio import gr, Block
 from safe_search import safe_search
 from i_search import google, i_search as i_s
-
 from agent import (
     ACTION_PROMPT,
     ADD_PROMPT,
@@ -23,12 +19,11 @@ from agent import (
     TASK_PROMPT,
     UNDERSTAND_TEST_RESULTS_PROMPT,
 )
-
 from utils import parse_action, parse_file_content, read_python_module_structure
 
-
-class App:
+class App(Block):
     def __init__(self):
+        super().__init__()
         self.app_state = {"components": []}
         self.terminal_history = ""
         self.components_registry = {
@@ -226,22 +221,7 @@ def run(self):
     print("Welcome to the Python App Builder!")
     print("Type 'help' to see the available commands.")
     print("-" * 50)
-    try:
-        while True:
-            try:
-                input_text = input("Enter input: ")
-            except EOFError:
-                print("Error: Input reading interrupted. Please provide valid input.")
-                continue
-
-            output, system_message = self.process_input(input_text)
-            if output:
-                print(output)
-            if system_message:
-                print(system_message)
-    except KeyboardInterrupt:
-        print("\nApplication stopped by user.")
-
+    self.launch()
 
 if __name__ == "__main__":
     app = App()