Mise à jour de l'interface utilisateur et des requêtes ALOQAS

app.py

@@ -1,10 +1,10 @@
 import streamlit as st
 import requests
-import streamlit as st
 import json
 
 siteUrl = "https://aloqas-aloqas-qa-fastapi.hf.space"
 
+response = None
 userContext = None
 context = None
 error = 0
@@ -13,13 +13,13 @@ limit = 9000
 with st.sidebar:
     st.title("Configuration")
     model = st.selectbox('Which model would you like to use?',
-                         ('SqueezeBERT', 'BERT', 'DeBERTa-v2'))
+                         ('SqueezeBERT', 'BERT', 'DeBERTa'))
     contextSelect = st.radio("Pick a context mode:", ["Text", "File"])
     if contextSelect == "File":
         userContext = st.file_uploader("Pick a file for the context", accept_multiple_files=True)
         context = "/uploadfile/"
     else:
-        userContext = st.text_area("write the context")
+        userContext = st.text_area("Write the context")
         context = "/contextText/"
 
 st.title("ALOQAS")
@@ -35,37 +35,42 @@ if "messages" not in st.session_state:
 
 if prompt := st.chat_input(placeholder="Write to ALOQAS..."):
     st.session_state.messages.append({"role": "user", "content": prompt})
-    params = {'texte': prompt, "model": model}
+
+    params = {'question': prompt, "model": model.lower()}
+
     if userContext != '' and len(userContext) <= limit:
         if contextSelect == "File" and userContext:
             # Préparation de la liste des fichiers pour l'envoi
-            files = [("file", (file.name, file, file.type)) for file in userContext]
-            response = requests.post(siteUrl + context, params=params, files=files)
+            files = [("files", (file.name, file, file.type)) for file in userContext]
+            response = requests.post(siteUrl + context, data=params, files=files)
+            st.session_state.messages.append({"role": "assistant", "content": json.loads(response.text).get("answer", "Sorry, I couldn't find an answer for your question.")})
         else:
             params["context"] = userContext
-            request_url = f"{siteUrl}/{model.lower()}/?context={userContext}&question={prompt}"
-            response = requests.post(request_url)
-            print(response.text)
-            print(response)
-            st.session_state.messages.append({"role": "assistant", "content": json.loads(response.text)["answer"]})
+            response = requests.post(siteUrl + context, json=params)
+            st.session_state.messages.append({"role": "assistant", "content": json.loads(response.text).get("answer", "Sorry, I couldn't find an answer for your question.")})
 
-    elif userContext == '' :
-        error = 1
-    else : 
-        error = 2
 
-    # st.write("Statut de la requête:", response.status_code)
-    # st.write("Réponse du serveur:", response.text)
+#     # st.write("Statut de la requête:", response.status_code)
+#     # st.write("Réponse du serveur:", response.text)
 
-    if userContext != '' and len(userContext) <= limit:
+    if response is None:
+        error = 3
+    if userContext != '' and len(userContext) <= limit and response is not None:
         with st.sidebar:
             st.title("Statistics on the last answer")
-            st.write(f"Score: {round(json.loads(response.text)['score'], 3)}")
-            st.write(f"Start: {json.loads(response.text)['start']}")
-            st.write(f"End: {json.loads(response.text)['end']}")
+            st.write(f"Score: {round(json.loads(response.text).get('score', 0), 2)}")
+            st.write(f"Start: {json.loads(response.text).get('start', 0)}")
+            st.write(f"End: {json.loads(response.text).get('end', 0)}")
+
+    elif userContext == '' :
+        error = 1
+    elif len(userContext) > limit:
+        error = 2
+    else:
+        error = 3
 
 for msg in st.session_state.messages:
-    st.chat_message(msg["role"]).write(msg['content'])
+    st.chat_message(msg.get("role", "assistant")).write(msg.get("content", "Hi, I'm ALOQAS. How can I help you?"))
 
 if error == 1:
     message_rouge = "⚠️ Please provide a context via the menu on your left."
@@ -75,3 +80,9 @@ elif error == 2:
     message_rouge = f"⚠️ The message you submitted was too long, please reload the conversation and submit something shorter than {limit} caracter."
     st.markdown(f'<div style="color: white; background-color: #ff4444; padding: 10px; border-radius: 5px;">{message_rouge}</div>', unsafe_allow_html=True)
     error = 0
+elif error == 3:
+    message_rouge = f"⚠️ An error occured, please try again."
+    st.markdown(f'<div style="color: white; background-color: #ff4444; padding: 10px; border-radius: 5px;">{message_rouge}</div>', unsafe_allow_html=True)
+    error = 0
+else:
+    st.write("")