Update app.py

app.py

@@ -115,7 +115,7 @@ def query(category, text):
     2. Generate a response
     """
     
-    category = Category[category]
+    #category = Category[category]
     
     chain = categories.category_modules[category].chain
     formatted_prompt = chain.prompt.format_prompt(
@@ -147,9 +147,7 @@ def parse(category, chatbot):
     """Takes the chatbot prompt and response and updates the extracted information"""
     global PARSING_REGEXP
 
-    if category not in Category.__members__:
-        # Handle the case when an invalid category is provided
-        return {"error": "Invalid category provided"}
+    
         
     if chatbot:
         if len(chatbot) > 0:
@@ -158,8 +156,12 @@ def parse(category, chatbot):
             answer = ""
     else:
         answer = ""
+    
+    if category not in Category.__members__:
+        # Handle the case when an invalid category is provided
+        answer="test"  
         
-    category = Category[category]
+    #category = Category[category]
     chain = categories.category_modules[category].chain
     yield {"status": "Parsing response..."}
     try: