Update main.py

main.py

@@ -77,7 +77,11 @@ async def chat(chat_request: ChatRequest):
         print(cleaned_initial_ai_response)
         print()
         
-        final_ai_response = cleaned_initial_ai_response.split('¿cuál e')[0].strip().replace('\n\n', '\n').replace('<|end|>', '').replace('<|user|>', '').replace('<|system|>', '').replace('<|assistant|>', '')
+        final_ai_response = cleaned_initial_ai_response.partition('¿Cuál es')[0].strip()
+        final_ai_response = final_ai_response.partition('¿Cuáles')[0].strip()        
+        
+        #final_ai_response = cleaned_initial_ai_response.split('¿cuál e')[0].strip().replace('\n\n', '\n').replace('<|end|>', '').replace('<|user|>', '').replace('<|system|>', '').replace('<|assistant|>', '')
+        
         print("final_ai_response after cleaning/splitting/partitioning '¿cuál e'")
         print(final_ai_response)
         print()