Update app.py

app.py

@@ -26,6 +26,37 @@ from agent import (
 
 from utils import parse_action, parse_file_content, read_python_module_structure
 
+def parse_action(response):
+    """Parses the NLP response for action and component information.
+
+    Args:
+        response (str): The NLP model's response.
+
+    Returns:
+        tuple: A tuple containing the parsed information:
+            - component_id (dict): The ID of the component, as an object with an '_id' attribute.
+            - action (str): The action to perform on the component (e.g., "update", "remove").
+            - property_name (str): The name of the property to modify.
+            - property_value (str): The new value for the property.
+    """
+    component_id = None
+    action = None
+    property_name = None
+    property_value = None
+
+    # Example parsing logic (adjust based on your NLP model's response format)
+    if "Component ID:" in response:
+        component_id = response.split("Component ID:")[1].split(",")[0].strip()
+    if "Action:" in response:
+        action = response.split("Action:")[1].split(",")[0].strip()
+    if "Property:" in response:
+        property_name = response.split("Property:")[1].split(",")[0].strip()
+    if "Value:" in response:
+        property_value = response.split("Value:")[1].split(",")[0].strip()
+
+    # Return an object with an '_id' attribute
+    return {"_id": component_id}, action, property_name, property_value
+
 class App:
     def __init__(self):
         self.app_state = {"components": []}
@@ -180,40 +211,35 @@ class App:
             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)
+                component = next((comp for comp in self.app_state["components"] if comp["id"] == component_id["_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"
+                        return self.update_app_canvas(), f"System: Updated property '{property_name}' of component with ID {component_id['_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['_id']}\n"
                     else:
                         return "", f"Error: Invalid action: {action}\n"
                 else:
-                    return "", f"Error: Component with ID {component_id} not found\n"
+                    return "", f"Error: Component with ID {component_id['_id']} not found\n"
             else:
                 return "", f"Error: Failed to parse action from NLP response\n"
 
     def build_app(self):
         with gr.Blocks() as demo:
-            self.components = []  # Store component objects
             for component in self.app_state["components"]:
                 component_type = component["type"]
                 properties = component["properties"]
                 
                 if component_type == "Button":
-                    button = gr.Button(value=properties["label"], variant="primary")
-                    self.components.append(button)  # Store the button object
+                    gr.Button(value=properties["label"], variant="primary")
                 elif component_type == "Text Input":
-                    textbox = gr.Textbox(label=properties["placeholder"])
-                    self.components.append(textbox)
+                    gr.Textbox(label=properties["placeholder"])
                 elif component_type == "Image":
-                    image = gr.Image(label=properties["alt"])
-                    self.components.append(image)
+                    gr.Image(label=properties["alt"])
                 elif component_type == "Dropdown":
-                    dropdown = gr.Dropdown(choices=properties["choices"], label="Dropdown")
-                    self.components.append(dropdown)
+                    gr.Dropdown(choices=properties["choices"], label="Dropdown")
 
             with gr.Tab("Terminal"):
                 gr.Markdown("## Terminal")