Update app.py

app.py

@@ -26,7 +26,6 @@ from agent import (
 
 from utils import parse_action, parse_file_content, read_python_module_structure
 
-
 class App:
     def __init__(self):
         self.app_state = {"components": []}
@@ -64,7 +63,7 @@ class App:
                 "description": "A dropdown menu for selecting options",
                 "code_snippet": "gr.Dropdown(choices={{choices}}, label='Dropdown')"
             }
-        }    
+        }
         self.nlp_model_names = [
             "google/flan-t5-small",
             "Qwen/CodeQwen1.5-7B-Chat-GGUF",
@@ -89,38 +88,19 @@ class App:
         else:
             return "NLP model not available."
 
-    class Component:
-        def __init__(self, type, properties=None, id=None):
-            self.id = id or random.randint(1000, 9999)
-            self.type = type
-            self.properties = properties or self.components_registry[type]["properties"].copy()
-
-        def to_dict(self):
-            return {
-                "id": self.id,
-                "type": self.type,
-                "properties": self.properties,
-            }
-
-        def render(self):
-            if self.type == "Dropdown":
-                self.properties["choices"] = str(self.properties["choices"]).replace("[", "").replace("]", "").replace("'", "")
-            return self.components_registry[self.type]["code_snippet"].format(**self.properties)
-
     def update_app_canvas(self):
         components_html = "".join([f"<div>Component ID: {component['id']}, Type: {component['type']}, Properties: {component['properties']}</div>" for component in self.app_state["components"]])
         return components_html
 
-    def add_component(self, component_type):
-        if component_type in self.components_registry:
-            new_component = self.Component(component_type)
-            self.app_state["components"].append(new_component.to_dict())
-            return (
-                self.update_app_canvas(),
-                f"System: Added component: {component_type}\n",
-            )
-        else:
-            return None, f"Error: Invalid component type: {component_type}\n"
+    def add_component(self, component_type, properties=None):
+        try:
+            new_component = {
+                "type": component_type,
+                "properties": properties or {}
+            }
+            self.app_state["components"].append(new_component)
+        except Exception as e:
+            print(f"Error adding component: {e}")
 
     def run_terminal_command(self, command, history):
         output = ""
@@ -170,7 +150,6 @@ class App:
         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):
@@ -195,7 +174,7 @@ class App:
             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
+            return output, ""
         else:
             model_index = random.randint(0, len(self.nlp_models)-1)
             response = self.get_nlp_response(input_text, model_index)
@@ -205,22 +184,16 @@ class App:
                 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",
-                        )
+                        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",
-                        )
+                        return self.update_app_canvas(), f"System: Removed component with ID {component_id}\n"
                     else:
-                        return None, f"Error: Invalid action: {action}\n"
+                        return "", f"Error: Invalid action: {action}\n"
                 else:
-                    return None, f"Error: Component with ID {component_id} not found\n"
+                    return "", f"Error: Component with ID {component_id} not found\n"
             else:
-                return None, f"Error: Failed to parse action from NLP response\n"
+                return "", f"Error: Failed to parse action from NLP response\n"
 
     def build_app(self):
         with gr.Blocks() as demo:
@@ -237,15 +210,15 @@ class App:
                 elif component_type == "Dropdown":
                     gr.Dropdown(choices=properties["choices"], label="Dropdown")
 
-            with gr.Tab("Components"):
-                gr.Markdown("## Available Components")
-                component_type = gr.Dropdown(label="Component Type", choices=list(self.components_registry.keys()))
-                properties = gr.JSON(label="Properties")
-                add_component_button = gr.Button("Add Component")
-                add_component_button.click(
-                    self.add_component,
-                    inputs=[component_type, properties],
-                    outputs=[]
+            with gr.Tab("Terminal"):
+                gr.Markdown("## Terminal")
+                terminal_input = gr.Textbox(label="Input")
+                terminal_output = gr.Textbox(label="Output")
+                run_terminal_command_button = gr.Button("Run Command")
+                run_terminal_command_button.click(
+                    self.run_terminal_command,
+                    inputs=[terminal_input],
+                    outputs=[terminal_output]
                 )
 
             with gr.Tab("NLP Models"):
@@ -258,20 +231,64 @@ class App:
                     inputs=[input_text, model_index],
                     outputs="text"
                 )
-            
-            with gr.Tab("Terminal"):
-                gr.Markdown("## Terminal")
-                terminal_input = gr.Textbox(label="Input")
-                terminal_output = gr.Textbox(label="Output")
-                run_terminal_command_button = gr.Button("Run Command")
-                run_terminal_command_button.click(
-                    self.run_terminal_command,
-                    inputs=[terminal_input],
-                    outputs=[terminal_output]
-                )
 
         return demo
 
+    def run(self):
+        self._print_welcome_message()
+        
+        while True:
+            try:
+                input_text = self._get_user_input()
+                if input_text.lower() == 'exit':
+                    break
+                elif input_text.lower() == 'launch':
+                    self.launch_app()
+                    continue
+                
+                output, system_message = self.process_input(input_text)
+                self._display_output(output, system_message)
+            
+            except EOFError:
+                print("Error: Input reading interrupted. Please provide valid input.")
+            except KeyboardInterrupt:
+                print("\nApplication stopped by user.")
+                break
+            except Exception as e:
+                print(f"An error occurred: {str(e)}")
+
+    def _print_welcome_message(self):
+        print("Welcome to the Python App Builder!")
+        print("Type 'help' to see the available commands.")
+        print("Type 'launch' to build
+        print("Type 'launch' to build and launch the Gradio app.")
+        print("Type 'exit' to quit the application.")
+        print("-" * 50)
+
+    def _get_user_input(self):
+        return input("Enter input: ").strip()
+
+    def _display_output(self, output, system_message):
+        if output:
+            print(output)
+        if system_message:
+            print(system_message)
+
+    def launch_app(self):
+        demo = self.build_app()
+        demo.launch()
+
+def main():
+    try:
+        app = App()
+        demo = app.build_app()
+        demo.launch()
+    except Exception as e:
+        print(f"Error launching app: {e}")
+
+if __name__ == "__main__":
+    main()
+
     def run(self):
         self._print_welcome_message()