Update app.py

app.py

@@ -1,11 +1,16 @@
-from transformers import pipeline
+import streamlit as st
+from main import generar_respuesta
 
-# Load the classification pipeline with the specified model
-pipe = pipeline("text-classification", model="tabularisai/multilingual-sentiment-analysis")
+st.title("📦 Asistente SeguroCargo")
+st.subheader("Consulta sobre tus envíos nacionales e internacionales")
+st.write("Bienbenido al chatbot de asistencion de SeguroCargo, cuentanos en que te podemos ayudar el dia de hoy")
 
-# Classify a new sentence
-sentence = "I love this product! It's amazing and works perfectly."
-result = pipe(sentence)
+pregunta = st.text_input("Haz tu pregunta:")
 
-# Print the result
-print(result)
+if st.button("Enviar"):
+    if pregunta.strip():
+        with st.spinner("Generando respuesta..."):
+            respuesta = generar_respuesta(pregunta)
+            st.success(respuesta)
+    else:
+        st.warning("Por favor, escribe una pregunta.")