Update app.py

app.py

@@ -68,14 +68,14 @@ def combined_function(user_prompt, similarity_threshold, selected_option, system
         if search_results:
             # Usar el primer resultado y dividirlo usando partial_message
             content = search_results[0]['content']
-            partial_message = content.split("\n")[0]  # Solo tomar la primera línea del contenido
+            partial_message = content.split("?")[0]  # Solo tomar la primera línea del contenido
             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("?")[0]  # Solo tomar la primera línea de la respuesta
+        partial_message = yi_coder_response.split("\n")[0]  # Solo tomar la primera línea de la respuesta
         return partial_message, None
     elif selected_option == "Ambos (basado en umbral de similitud)":
         # Realizar búsqueda vectorial