Update app.py

app.py

@@ -2,25 +2,54 @@ import streamlit as st
 import requests
 import os
 import json
+from dotenv import load_dotenv
+import logging
+from dotenv import load_dotenv
+from requests.adapters import HTTPAdapter
+from requests.packages.urllib3.util.retry import Retry
+
+# Configurar logging
+logging.basicConfig(level=logging.INFO)
+
+# Cargar variables de entorno
+load_dotenv()
 
 # Configuración de la API de Flowise
-API_URL = os.getenv("API_URL", "https://livanarzuaga-test.hf.space/api/v1/prediction/2a2e535d-1807-4701-8468-c0ede26bcdbf")
+API_URL = os.getenv("API_URL", "https://livanarzuaga-test.hf.space/api/v1/prediction/2dba91d7-15b7-4475-b0ec-34a11ffbeab5")
+session = requests.Session()
+
+# Configurar reintentos
+retries = Retry(total=5, backoff_factor=1, status_forcelist=[502, 503, 504])
+session.mount('https://', HTTPAdapter(max_retries=retries))
 
 def query(payload, api_url=API_URL):
     try:
-        response = requests.post(api_url, json=payload)
+        response = session.post(api_url, json=payload)
         response.raise_for_status()
         return response.json()
     except requests.exceptions.HTTPError as http_err:
-        return {"error": f"HTTP error occurred: {http_err}"}
+        logging.error(f"HTTP error occurred: {http_err}")
+        return {"error": "HTTP error occurred. Please try again later."}
     except requests.exceptions.ConnectionError as conn_err:
-        return {"error": f"Connection error occurred: {conn_err}"}
+        logging.error(f"Connection error occurred: {conn_err}")
+        return {"error": "Connection error occurred. Please try again later."}
     except requests.exceptions.Timeout as timeout_err:
-        return {"error": f"Timeout error occurred: {timeout_err}"}
+        logging.error(f"Timeout error occurred: {timeout_err}")
+        return {"error": "Timeout error occurred. Please try again later."}
     except requests.exceptions.RequestException as req_err:
-        return {"error": f"An error occurred: {req_err}"}
+        logging.error(f"An error occurred: {req_err}")
+        return {"error": "An error occurred. Please try again later."}
+
+def validate_input(user_input):
+    if not user_input or len(user_input.strip()) == 0:
+        return False, "Input cannot be empty."
+    return True, ""
 
 def get_response(user_input, api_url=API_URL):
+    is_valid, error_message = validate_input(user_input)
+    if not is_valid:
+        return error_message
+    
     payload = {"question": user_input}
     response = query(payload, api_url)
     return response.get('text', "Error: No API response.")
@@ -35,6 +64,7 @@ def export_conversation(past, generated, file_path="conversation.json"):
             json.dump(conversation, f)
         return True
     except Exception as e:
+        logging.error(f"Error exporting conversation: {e}")
         return str(e)
 
 # Inicializar claves en st.session_state