Clear thread functionality

app.py

@@ -1,5 +1,4 @@
 import os
-import urllib
 import urllib.request
 from time import sleep
 from typing import Dict, List, Generator
@@ -15,43 +14,38 @@ class MockInterviewer:
     def __init__(self) -> None:
         self._client = openai.OpenAI(api_key=os.environ['OPENAI_API_KEY'])
         self._assistant_id_cache: Dict[str, str] = {}
+        self.clear_thread()
 
     def chat(self, usr_message: Dict, history: List[List], job_role: str, company: str) -> Generator:
-        print('Started function')
+        print('Started chat')
         self._validate_fields(job_role, company)
-
-        thread = self._client.beta.threads.create()
-        user_input = usr_message.get('text')
-
         assistant_id = self._init_assistant(job_role, company)
+        yield self._send_message(usr_message.get('text'), assistant_id)
 
-        # Add the user's message to the thread
-        self._client.beta.threads.messages.create(thread_id=thread.id,
-                                                 role="user",
-                                                 content=user_input)
-        print('Client made')
-        # Run the Assistant
-        run = self._client.beta.threads.runs.create(thread_id=thread.id,
-                                                   assistant_id=assistant_id)
+    def clear_thread(self) -> None:
+        print('Initializing new thread')
+        self._thread = self._client.beta.threads.create()
+
+    def _send_message(self, message: str, assistant_id: str) -> str:
+        self._client.beta.threads.messages.create(thread_id=self._thread.id, role='user', content=message)
+        print('Message created')
+        run = self._client.beta.threads.runs.create(thread_id=self._thread.id, assistant_id=assistant_id)
         print('Run created')
 
         # Check if the Run requires action (function call)
         while True:
-            run_status = self._client.beta.threads.runs.retrieve(thread_id=thread.id,
-                                                                run_id=run.id)
-            print(f"Run status: {run_status.status}")
+            run_status = self._client.beta.threads.runs.retrieve(thread_id=self._thread.id, run_id=run.id)
+            print(f'Run status: {run_status.status}')
             if run_status.status == 'completed':
                 break
 
             sleep(1)  # Wait for a second before checking again
 
         # Retrieve and return the latest message from the assistant
-        messages = self._client.beta.threads.messages.list(thread_id=thread.id)
+        messages = self._client.beta.threads.messages.list(thread_id=self._thread.id)
         response = messages.data[0].content[0].text.value
-
-        print(f"Assistant response: {response}")  # Debugging line
-        #return json.dumps({"response": response})
-        yield response
+        print(f'Assistant response: {response}')
+        return response
 
     def _validate_fields(self, job_role: str, company: str) -> None:
         if not job_role and not company:
@@ -105,7 +99,7 @@ class MockInterviewer:
 with gr.Blocks() as demo:
     mock_interviewer = MockInterviewer()
 
-    chatbot = gr.ChatInterface(
+    chat_interface = gr.ChatInterface(
         mock_interviewer.chat,
         additional_inputs=[
             gr.Textbox(label='Job Role', placeholder='Product Manager'),
@@ -116,6 +110,9 @@ with gr.Blocks() as demo:
         multimodal=True,
         retry_btn=None,
         undo_btn=None).queue()
+    
+    chat_interface.load(mock_interviewer.clear_thread)
+    chat_interface.clear_btn.click(mock_interviewer.clear_thread)
 
 if __name__ == '__main__':
     demo.launch().queue()