Update app.py

app.py

@@ -27,6 +27,7 @@ def res(prompt, historial):
 
     return respuesta, historial
 
+
 def respond(message, chat_history, history):
     response = res(message, history)
     bot_message = response[0]
@@ -35,19 +36,30 @@ def respond(message, chat_history, history):
 
     # Check if the conversation is over
     if "vista" in bot_message.lower():  # Check if "vista" is in the bot's response
-        chat_history.append(("System", "The word 'vista' was found in the bot's response."))  
+        #chat_history.append(("System", "The word 'vista' was found in the bot's response."))  
         # Get the assistant's last message
         last_message = response[1][-1]['content']  # assuming the last message is from the assistant
-        #chat_history.append(("System", last_message)) 
-        # Pass the JSON output to the extract_info function
-        # info = extract_info(last_message)
-        # Use the extracted info
-        #WA(info)
-        #print(last_message)
+        
+        # Split the last message by "vista."
+        parts = last_message.split("vista.")
+
+        # If there are two parts (text + JSON), process the JSON part
+        if len(parts) == 2:
+            json_part = parts[1].strip()  # Extract the JSON part and remove leading/trailing spaces
+            # Now you can pass the JSON output to the extract_info function
+            info = extract_info(json_part)
+            # Use the extracted info or append it to chat_history, etc.
+            chat_history.append(("System", json_part))  # Example: append the extracted JSON to chat history
+            # WA(info)
+            #print(info)  # Example: print the extracted info
+
+        chat_history.append(("System", parts[0]))  # Append the text part to chat history
 
     return "", chat_history, response[1]
 
 
+
+
 def WA(history):
   sendWA("491771675218",history)
   return "✅ Validado"