Update app.py

app.py

@@ -58,6 +58,12 @@ def get_conversational_chain():
     prompt = PromptTemplate(template=prompt_template, input_variables=["context", "question"])
     chain = load_qa_chain(model, chain_type="stuff", prompt=prompt)
     return chain
+    
+def eliminar_texto_entre_tags(texto):
+    patron = r'<think>.*?</think>'
+    texto_limpio = re.sub(patron, '', texto)
+    return texto_limpio
+
 
 def user_input(user_question):
     """Handles user queries by retrieving answers from the vector store."""
@@ -85,7 +91,7 @@ def user_input(user_question):
             thought_process = thought_process_match.group(1).strip()
 
     # Remove the thought process from the main response
-    clean_response = original_response.split("<think>")[-1].split("</think>")[0]
+    clean_response = eliminar_texto_entre_tags(original_response)
 
     # Debugging: Print the cleaned response
     print("Cleaned Response:", clean_response)