Update app.py

app.py

@@ -41,6 +41,11 @@ def process_pdfs_in_folder(folder_path):
             pdf_texts[filename] = text
     return pdf_texts
 
+# Function to get a response related to reference information
+def get_reference_response():
+    # Replace this with logic to generate a response based on reference information
+    return "Lo siento, no puedo proporcionar una respuesta específica, pero aquí hay información de referencia relevante."
+
 # Streamlit app
 def main():
     st.set_page_config(page_title="MAX Chatbot - INIF", page_icon="🤖")
@@ -130,14 +135,12 @@ def main():
                     if response:
                         if response.candidates:
                             parts = response.candidates[0].content.parts
-                            generated_text = parts[0].text if parts else "No se generó contenido."
+                            generated_text = parts[0].text if parts else get_reference_response()
                             st.markdown(f"Assistant: {to_markdown(generated_text)}")
                             messages.append({"role": "model", "parts": [generated_text]})
                         else:
-                            # 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]})
+                            st.markdown(f"Assistant: {to_markdown(get_reference_response())}")
+                            messages.append({"role": "model", "parts": [get_reference_response()]})
             else:
                 # Otros modelos Gemini seleccionados
                 response = get_response(user_input_with_context)
@@ -146,11 +149,9 @@ def main():
                 res_text = ""
                 for chunk in response:
                     res_text += chunk.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]})
+                if not res_text.strip():
+                    st.markdown(f"Assistant: {to_markdown(get_reference_response())}")
+                    messages.append({"role": "model", "parts": [get_reference_response()]})
                 else:
                     st.markdown(f"Assistant: {to_markdown(res_text)}")
                     messages.append({"role": "model", "parts": [res_text]})