fix chat issue

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import requests
 import streamlit as st
+import json
 
 siteUrl = "https://aloqas-aloqas-qa-fastapi.hf.space"
 
@@ -29,12 +30,8 @@ if "messages" not in st.session_state:
         {"role": "assistant", "content": "Hi, I'm ALOQAS. How can I help you?"}
     ]
 
-for msg in st.session_state.messages:
-    st.chat_message(msg["role"]).write(msg["content"])
-
 if prompt := st.chat_input(placeholder="Write to ALOQAS..."):
     st.session_state.messages.append({"role": "user", "content": prompt})
-    st.chat_message("user").write(prompt)
     params = {'texte': prompt, "model": model}
     if userContext is not None:
         if contextSelect == "File" and userContext:
@@ -43,9 +40,15 @@ if prompt := st.chat_input(placeholder="Write to ALOQAS..."):
             response = requests.post(siteUrl + context, params=params, files=files)
         else:
             params["context"] = userContext
-            response = requests.post(siteUrl + context, params=params)
+            request_url = f"{siteUrl}/{model.lower()}/?context={userContext}&question={prompt}"
+            response = requests.post(request_url)
+            print(response.text)
+            st.session_state.messages.append({"role": "assistant", "content": json.loads(response.text)["answer"]})
+
     else:
         response = requests.post(siteUrl + '/withoutFile/', params=params)
 
-    st.write("Statut de la requête:", response.status_code)
-    st.write("Réponse du serveur:", response.text)
+for msg in st.session_state.messages:
+    st.chat_message(msg["role"]).write(msg["content"])
+    # st.write("Statut de la requête:", response.status_code)
+    # st.write("Réponse du serveur:", response.text)