Update app.py

app.py

@@ -149,49 +149,46 @@ def procesar_consulta(user_question):
     respuesta_final = eliminar_proceso_pensamiento(response['output_text'])
     mostrar_respuesta(respuesta_final)
 
-from googletrans import Translator
-
-def traducir_contexto(contexto):
-    translator = Translator()
-    return translator.translate(contexto, src='en', dest='es').text
-
 def generar_sugerencias():
-    """Genera preguntas sugeridas breves y directas en español"""
+    """Genera preguntas sugeridas usando los chunks más relevantes del documento"""
     if 'vector_store' not in st.session_state:
         return []
     
     try:
+        # Obtener los fragmentos más relevantes
         docs = st.session_state.vector_store.similarity_search("", k=3)
         context = "\n".join([doc.page_content for doc in docs])
         
-        # Traducir el contexto al español si está en otro idioma
-        context_es = traducir_contexto(context)
-        
+        # Prompt para generación de sugerencias
         prompt_template = """
-        Genera exactamente 3 preguntas muy breves (5-8 palabras) sobre los puntos clave.
-        Las preguntas deben ser simples y estar redactadas en español.
-        Formato: Lista numerada sin explicaciones.
+        Basado en el siguiente contexto, genera exactamente 3 preguntas relevantes en español.
+        Formato: Cada pregunta debe ser concisa y específica.
+        Las preguntas deben ser útiles para entender el contenido del documento.
         
         Contexto:
         {context}
+        
+        Preguntas sugeridas:
+        1.
         """
         
         model = ChatGroq(
-            temperature=0.3,
+            temperature=0.7,
             model_name="deepseek-r1-distill-llama-70b",
             groq_api_key=os.getenv("GROQ_API_KEY")
         )
         
-        response = model.invoke(prompt_template.format(context=context_es))
+        response = model.invoke(prompt_template.format(context=context))
         
+        # Procesar la respuesta para extraer las preguntas
         preguntas = []
         for line in response.content.split("\n"):
-            line = line.strip()
-            if line and line[0].isdigit():
-                pregunta = line.split('. ', 1)[1] if '. ' in line else line[2:]
-                preguntas.append(pregunta[:60])  # Limitar longitud máxima
+            if line.strip() and any(c.isdigit() for c in line[:3]):
+                pregunta = line.split('.', 1)[1].strip() if '.' in line else line.strip()
+                if pregunta:
+                    preguntas.append(pregunta)
         
-        return preguntas[:3]
+        return preguntas[:3]  # Asegurar máximo 3 preguntas
         
     except Exception as e:
         st.error(f"Error generando sugerencias: {str(e)}")
@@ -299,4 +296,4 @@ def main():
         st.write("Por favor, sube un documento para continuar.")
 
 if __name__ == "__main__":
-    main()
+    main()