Update app.py

app.py

@@ -62,20 +62,26 @@ def generate_code(system_prompt, user_prompt, max_length):
 # Función para combinar búsqueda vectorial y Yi-Coder
 @gpu_decorator(duration=100)
 def combined_function(user_prompt, similarity_threshold, selected_option, system_prompt, max_length):
+    def get_partial_message(response):
+        """Obtiene el contenido después de 'Respuesta:' si está presente en la respuesta."""
+        if "Respuesta:" in response:
+            return response.split("Respuesta:")[1].strip()  # Tomar solo el texto después de 'Respuesta:'
+        else:
+            return response  # Devolver la respuesta completa si no contiene 'Respuesta:'
+
     if selected_option == "Solo Búsqueda Vectorial":
         # Realizar búsqueda vectorial
         search_results = vector_search(user_prompt, embedding_model, index)
         if search_results:
-            # Usar el primer resultado y dividirlo usando partial_message
             content = search_results[0]['content']
-            partial_message = content.split("?")[0]  # Solo tomar la primera línea del contenido
+            partial_message = get_partial_message(content)
             return partial_message, None
         else:
             return "No se encontraron resultados en Pinecone.", None
     elif selected_option == "Solo Yi-Coder":
         # Generar respuesta usando Yi-Coder
         yi_coder_response = generate_code(system_prompt, user_prompt, max_length)
-        partial_message = yi_coder_response.split("\n")[0]  # Solo tomar la primera línea de la respuesta
+        partial_message = get_partial_message(yi_coder_response)
         return partial_message, None
     elif selected_option == "Ambos (basado en umbral de similitud)":
         # Realizar búsqueda vectorial
@@ -84,15 +90,15 @@ def combined_function(user_prompt, similarity_threshold, selected_option, system
             top_result = search_results[0]
             if top_result['score'] >= similarity_threshold:
                 content = top_result['content']
-                partial_message = content.split("\n")[0]  # Solo tomar la primera línea del contenido
+                partial_message = get_partial_message(content)
                 return partial_message, None
             else:
                 yi_coder_response = generate_code(system_prompt, user_prompt, max_length)
-                partial_message = yi_coder_response.split("\n")[0]  # Solo tomar la primera línea de la respuesta
+                partial_message = get_partial_message(yi_coder_response)
                 return partial_message, None
         else:
             yi_coder_response = generate_code(system_prompt, user_prompt, max_length)
-            partial_message = yi_coder_response.split("\n")[0]  # Solo tomar la primera línea de la respuesta
+            partial_message = get_partial_message(yi_coder_response)
             return partial_message, None
     else:
         return "Opción no válida.", None