Update app.py

app.py

@@ -3,6 +3,7 @@ import subprocess
 import random
 import json
 from datetime import datetime
+import uuid  # Import the uuid library
 
 from huggingface_hub import InferenceClient, cached_download, hf_hub_url
 import gradio as gr
@@ -26,37 +27,6 @@ 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": []}
@@ -127,7 +97,8 @@ class App:
         try:
             new_component = {
                 "type": component_type,
-                "properties": properties or {}
+                "properties": properties or {},
+                "id": str(uuid.uuid4())  # Assign a unique ID using UUID
             }
             self.app_state["components"].append(new_component)
         except Exception as e:
@@ -211,18 +182,18 @@ 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["_id"]), None)
+                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['_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['_id']}\n"
+                        return self.update_app_canvas(), f"System: Removed component with ID {component_id}\n"
                     else:
                         return "", f"Error: Invalid action: {action}\n"
                 else:
-                    return "", f"Error: Component with ID {component_id['_id']} not found\n"
+                    return "", f"Error: Component with ID {component_id} not found\n"
             else:
                 return "", f"Error: Failed to parse action from NLP response\n"