Update app.py

app.py

@@ -1,61 +1,3 @@
-import gradio as gr
-import os
-import yaml
-
-# Function to update the NLU file with new intents
-def update_nlu_file(intent_name, training_examples, nlu_file_path="data/nlu.yml"):
-    # Create the 'data' directory if it doesn't exist
-    if not os.path.exists("data"):
-        os.makedirs("data")
-
-    # If the NLU file doesn't exist, create it with a default structure
-    if os.path.exists(nlu_file_path):
-        with open(nlu_file_path, "r") as file:
-            nlu_data = yaml.safe_load(file)
-    else:
-        nlu_data = {"rasa_nlu_data": {"common_examples": []}}
-
-    # Add the new intent and examples
-    new_intent = {
-        "intent": intent_name,
-        "examples": training_examples.splitlines()
-    }
-
-    nlu_data["rasa_nlu_data"]["common_examples"].append(new_intent)
-
-    # Write back the updated NLU file
-    with open(nlu_file_path, "w") as file:
-        yaml.dump(nlu_data, file)
-
-    return f"NLU file updated with intent: {intent_name}"
-
-# Function to update the Story file with new stories
-def update_stories_file(story_name, steps, stories_file_path="data/stories.yml"):
-    # Create the 'data' directory if it doesn't exist
-    if not os.path.exists("data"):
-        os.makedirs("data")
-
-    # If the Stories file doesn't exist, create it with a default structure
-    if os.path.exists(stories_file_path):
-        with open(stories_file_path, "r") as file:
-            stories_data = yaml.safe_load(file)
-    else:
-        stories_data = {"stories": []}
-
-    # Add the new story
-    new_story = {
-        "story": story_name,
-        "steps": [{"intent": step} for step in steps.splitlines()]
-    }
-
-    stories_data["stories"].append(new_story)
-
-    # Write back the updated Stories file
-    with open(stories_file_path, "w") as file:
-        yaml.dump(stories_data, file)
-
-    return f"Stories file updated with story: {story_name}"
-
 # Function to update the Domain file with new intents and responses
 def update_domain_file(intent_name, domain_file_path="domain.yml"):
     # Create the 'data' directory if it doesn't exist
@@ -66,6 +8,9 @@ def update_domain_file(intent_name, domain_file_path="domain.yml"):
     if os.path.exists(domain_file_path):
         with open(domain_file_path, "r") as file:
             domain_data = yaml.safe_load(file)
+            # If domain_data is None, initialize it with a default structure
+            if domain_data is None:
+                domain_data = {"intents": [], "responses": {}}
     else:
         domain_data = {"intents": [], "responses": {}}
 
@@ -84,32 +29,3 @@ def update_domain_file(intent_name, domain_file_path="domain.yml"):
         yaml.dump(domain_data, file)
 
     return f"Domain file updated with intent: {intent_name}"
-
-# Gradio interface function
-def rasa_creator(intent_name, training_examples, story_name, steps, add_button):
-    if add_button:
-        nlu_response = update_nlu_file(intent_name, training_examples)
-        stories_response = update_stories_file(story_name, steps)
-        domain_response = update_domain_file(intent_name)
-        return nlu_response, stories_response, domain_response
-    return "Press Add to update files."
-
-# Gradio interface
-iface = gr.Interface(
-    fn=rasa_creator,
-    inputs=[
-        gr.Textbox(label="Intent Name"),
-        gr.Textbox(label="Training Examples", placeholder="Add your training examples here", lines=4),
-        gr.Textbox(label="Story Name"),
-        gr.Textbox(label="Story Steps", placeholder="Add your story steps here", lines=4),
-        gr.Button("Add")  # Button to trigger the file update
-    ],
-    outputs=[
-        gr.Textbox(label="NLU Update Status"),
-        gr.Textbox(label="Stories Update Status"),
-        gr.Textbox(label="Domain Update Status")
-    ],
-    live=False,  # Disable live mode to prevent auto-updating
-)
-
-iface.launch()