Update app.py

app.py

@@ -134,7 +134,10 @@ def main():
                             st.markdown(f"Assistant: {to_markdown(generated_text)}")
                             messages.append({"role": "model", "parts": [generated_text]})
                         else:
-                            st.warning("No se encontraron candidatos en la respuesta.")
+                            # Responder con información de referencia si no hay respuesta basada en el contexto
+                            st.warning("No se encontraron candidatos en la respuesta. Aquí hay información de referencia:")
+                            st.markdown(f"Assistant: {to_markdown(inif_context)}")
+                            messages.append({"role": "model", "parts": [inif_context]})
             else:
                 # Otros modelos Gemini seleccionados
                 response = get_response(user_input_with_context)
@@ -143,8 +146,14 @@ def main():
                 res_text = ""
                 for chunk in response:
                     res_text += chunk.text
-                st.markdown(f"Assistant: {to_markdown(res_text)}")
-                messages.append({"role": "model", "parts": [res_text]})
+                if not res_text:
+                    # Responder con información de referencia si no hay respuesta basada en el contexto
+                    st.warning("No se encontró respuesta basada en el contexto. Aquí hay información de referencia:")
+                    st.markdown(f"Assistant: {to_markdown(inif_context)}")
+                    messages.append({"role": "model", "parts": [inif_context]})
+                else:
+                    st.markdown(f"Assistant: {to_markdown(res_text)}")
+                    messages.append({"role": "model", "parts": [res_text]})
 
     # Actualizar historial de mensajes en la sesión de Streamlit
     st.session_state["messages"] = messages