Update app.py

app.py

@@ -3,11 +3,10 @@ import streamlit as st
 import os
 import time
 from dotenv import load_dotenv
-
 load_dotenv()
 st.set_page_config(page_title="Pak Case Law ChatBot")
 st.header("Welcome To Pakistani AI Lawyer!")
-st.markdown("Currently ChatBot is taking content only from this [file.](http://kpcode.kp.gov.pk/uploads/1964_35_THE_WEST_PAKISTAN_FAMILY_COURTS_ACT_1964.pdf)")
+st.markdown("Currently the ChatBot is taking content only from this [file.](http://kpcode.kp.gov.pk/uploads/1964_35_THE_WEST_PAKISTAN_FAMILY_COURTS_ACT_1964.pdf)")
 
 class OpenAIClient:
     def __init__(self):
@@ -15,30 +14,31 @@ class OpenAIClient:
         self.thread_id = None
         self.file_id = ["file-Anvyhz2k0puV0NCjWH8mGsuk"]
         self.assistant_id = "asst_Z7pj5QGUjiAlVpJzqsX2SJkg"
-        self.message_id = "msg_Ym6yy4hm2Anai4Snv2xaCX35"
         self.run_id = "run_A5wq14ZMQXBpaDnULFPky5Bb"
-        self.thread_id = "thread_twTf1EOyZPYMHnCNbA5NA5Bw"
 
+    def create_thread_id(self):
+        if "thread_id" not in st.session_state:
+            _id = self.client.beta.threads.create().id
+            st.session_state["thread_id"] = _id
+        
     def main(self, user_message):
-        lawyer = self.client.beta.assistants.retrieve(self.assistant_id)
         message = self.client.beta.threads.messages.create(
-            self.thread_id,
+            st.session_state.get("thread_id"),
             role="user",
             content=user_message,
         )
-        _run = self.client.beta.threads.runs.create(thread_id=self.thread_id, assistant_id=self.assistant_id)
+        _run = self.client.beta.threads.runs.create(thread_id=st.session_state.get("thread_id"), assistant_id=self.assistant_id)
         while True:
-            run = self.client.beta.threads.runs.retrieve(thread_id=self.thread_id,
+            run = self.client.beta.threads.runs.retrieve(thread_id=st.session_state.get("thread_id"),
                                                         run_id=_run.id).status
 
-            time.sleep(5)
+            time.sleep(3)
             if run == "completed":
                 return self.get_response()
             elif run == "queued":
                 continue
             elif run == "failed" or run == "cancelled" or run == "expired":
-                print(run)
-                break
+                return run
             elif run == "in_progress":
                 continue
             else:
@@ -46,27 +46,21 @@ class OpenAIClient:
 
     def get_response(self):
         messages = self.client.beta.threads.messages.list(
-            thread_id=self.thread_id
+            thread_id=st.session_state.get("thread_id")
         )
         return messages.data[0].content[0].text.value
 
-    def file_handler(self):
-        return self.client.files.retrieve("file-CwRdM2JgGKEVZP4rhoKSvgpD")
-
-    def delete_file(self):
-        return self.client.files.list()
-
 try:
     client = OpenAIClient()
+    client.create_thread_id()
     user_input = st.text_input("Input: ", key='input')
-    response = client.main(user_input)
     submit = st.button("Ask the Question")
     if submit:
         st.subheader("The Response is")
-        st.write(response)
+        st.write(client.main(user_input))
 except openai.AuthenticationError:
     st.write("Your API key or token was invalid, expired, or revoked.")
 except openai.RateLimitError:
     st.write("You have hit your assigned rate limit. Please try again in a few minutes.")
 except Exception as e:
-    st.write(e)
+    st.write(e)