Update app.py

app.py

@@ -1,65 +1,122 @@
 import streamlit as st
-from streamlit_chat import message
 import requests
 import os
+import json
+from dotenv import load_dotenv
+import logging
+from requests.adapters import HTTPAdapter
+from requests.packages.urllib3.util.retry import Retry
 
-# FlowiseAI API configuration
+# 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")
 
-def query(payload):
-    response = requests.post(API_URL, json=payload)
-    
-    # Check if the response is empty or not valid JSON
-    if response.status_code != 200 or not response.content.strip():
-        return {}
+#API_URL = os.getenv("API_URL", "https://wilkie-hf-space-tuto.hf.space/api/v1/prediction/037dd3d4-8492-45c4-9c5f-6ff84b8e95ee")
+
+#API_URL = os.getenv("API_URL", "https://dsalazarcazanhas-flowise.hf.space/api/v1/prediction/9310432d-f1c9-4194-80a4-84bd3d8ac6af")
+#head = {"Authorization": "Bearer -rejhzEZZ2xaG1P8zGfKYnRBy-wJXpbbE13ld-ZaFEY"}
+
+#API_URL = os.getenv("API_URL","https://marie0501-boulder-weather-flowise.hf.space/api/v1/prediction/dd2d1375-8019-4f52-bc23-1d0845b4a0c1")
+#head = {"Authorization": "Bearer o6Xy3bIuPhDJV_z0PefeuvbHO2X3yo3Rt7NSiW-bY4o"}
 
+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 = session.post(api_url, json=payload)
+        response.raise_for_status()
         return response.json()
-    except requests.exceptions.JSONDecodeError:
-        # Handle cases where the response body is not valid JSON
-        return {}
-
-
-def get_flowise_response(user_input):
-    payload = {
-        "question": user_input,
-        "overrideConfig": {
-        "sessionId": "user1"
-    }
-    }
-    response = query(payload)
+    except requests.exceptions.HTTPError as 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:
+        logging.error(f"Connection error occurred: {conn_err}")
+        return {"error": "Connection error occurred. Please try again later."}
+    except requests.exceptions.Timeout as 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:
+        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, history, api_url=API_URL):
+    is_valid, error_message = validate_input(user_input)
+    if not is_valid:
+        return error_message
     
-    if 'text' in response:
-        return response['text']
-    return "Error: Unable to get a response from the API."
+    # Incluir el historial completo en el payload
+    payload = {"question": user_input, "history": history, "overrideConfig": {"sessionId": "user1"}}
+    response = query(payload, api_url)
+    return response.get('text', "Error: No API response.")
 
+def export_conversation(past, generated, file_path="conversation.json"):
+    try:
+        conversation = {
+            "past": past,
+            "generated": generated
+        }
+        with open(file_path, "w") as f:
+            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
 if 'history' not in st.session_state:
     st.session_state['history'] = []
 
 if 'generated' not in st.session_state:
-    st.session_state['generated'] = ["Hello! Ask me anything about the weather or any other topic 🤖"]
+    st.session_state['generated'] = ["Hi I can answer your weather related questions"]
 
 if 'past' not in st.session_state:
-    st.session_state['past'] = ["Hey! 👋"]
-
-# Container for the chat history
-response_container = st.container()
-# Container for the user's text input
-container = st.container()
-
-with container:
-    with st.form(key='my_form', clear_on_submit=True):
-        user_input = st.text_input("Query:", placeholder="Talk to the chatbot here (:", key='input')
-        submit_button = st.form_submit_button(label='Send')
-        
+    st.session_state['past'] = ["Hello"]
+
+# Sidebar con opciones
+st.sidebar.title("Opciones")
+if st.sidebar.button("Nueva conversación"):
+    st.session_state['past'] = ["Hello"]
+    st.session_state['generated'] = ["Hi I can answer your weather related questions"]
+    st.session_state['history'] = []
+    st.sidebar.success("Nueva conversación iniciada")
+
+if st.sidebar.button("Exportar conversación"):
+    result = export_conversation(st.session_state['past'], st.session_state['generated'])
+    if result is True:
+        st.sidebar.success("Conversación exportada como conversation.json")
+    else:
+        st.sidebar.error(f"Error al exportar la conversación: {result}")
+
+# Contenedor para el historial del chat
+chat_container = st.container()
+
+# Contenedor para la entrada de texto del usuario
+with st.form(key='my_form', clear_on_submit=True):
+    user_input = st.text_input("Chat:", placeholder="Type here to chat", key='input')
+    submit_button = st.form_submit_button(label='Send')
+    
     if submit_button and user_input:
-        output = get_flowise_response(user_input)
-        
+        output = get_response(user_input, st.session_state['history'])
         st.session_state['past'].append(user_input)
         st.session_state['generated'].append(output)
+        st.session_state['history'].append({"user": user_input, "bot": output})
 
-if st.session_state['generated']:
-    with response_container:
-        for i in range(len(st.session_state['generated'])):
-            message(st.session_state["past"][i], is_user=True, key=str(i) + '_user', avatar_style="big-smile")
-            message(st.session_state["generated"][i], key=str(i), avatar_style="thumbs")
+# Mostrar el historial del chat
+with chat_container.container():
+    for i in range(len(st.session_state['generated'])):
+        st.write(f"**User:** {st.session_state['past'][i]}")
+        st.write(f"**Bot:** {st.session_state['generated'][i]}")