Update app.py

app.py

@@ -66,15 +66,17 @@ def combined_function(user_prompt, similarity_threshold, selected_option, system
         # Realizar búsqueda vectorial
         search_results = vector_search(user_prompt, embedding_model, index)
         if search_results:
-            # Usar el primer resultado
+            # Usar el primer resultado y dividirlo usando partial_message
             content = search_results[0]['content']
-            return content, None
+            partial_message = content.split("\n")[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)
-        return yi_coder_response, None
+        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
         search_results = vector_search(user_prompt, embedding_model, index)
@@ -82,16 +84,20 @@ 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']
-                return content, None
+                partial_message = content.split("\n")[0]  # Solo tomar la primera línea del contenido
+                return partial_message, None
             else:
                 yi_coder_response = generate_code(system_prompt, user_prompt, max_length)
-                return yi_coder_response, None
+                partial_message = yi_coder_response.split("\n")[0]  # Solo tomar la primera línea de la respuesta
+                return partial_message, None
         else:
             yi_coder_response = generate_code(system_prompt, user_prompt, max_length)
-            return yi_coder_response, None
+            partial_message = yi_coder_response.split("\n")[0]  # Solo tomar la primera línea de la respuesta
+            return partial_message, None
     else:
         return "Opción no válida.", None
 
+
 # Funciones para el procesamiento de entradas y actualización de imágenes
 def process_input(message, history, selected_option, similarity_threshold, system_prompt, max_length):
     response, image = combined_function(message, similarity_threshold, selected_option, system_prompt, max_length)