Update app.py

app.py

@@ -7,35 +7,52 @@ print(gr.__version__)
 
 theme = MonolithTheme()
 
+
+
 # Initialize the OpenAI client once and use it in the openai_chatbot function
 api_key = os.getenv("OPENAI_API_KEY")
 client = openai.OpenAI(api_key=api_key)
 
-def openai_chatbot(message, history=None):
+assistant_id = 'asst_wsFyYTDSGAkPibNdRGBY1Sa1' # Replace with new assistant ID if you have a different one
+
+def openai_chatbot(message, history=None, thread_id=None):
     try:
-        conversation_history = history or []
-        if message:  # Only append if there is a message to send
-            conversation_history.append({"role": "user", "content": message})
+        if not thread_id:
+            # Create a new thread if one does not exist
+            thread = client.beta.threads.create()
+            thread_id = thread.id
+        
+        if message:  # Add the user's message to the thread
+            client.beta.threads.messages.create(
+                thread_id=thread_id,
+                role="user",
+                content=message
+            )
         
-        chat_completion = client.chat.completions.create(
-            model="gpt-3.5-turbo",
-            messages=conversation_history,
+        # Run the assistant to get the response
+        run = client.beta.threads.runs.create(
+            thread_id=thread_id,
+            assistant_id=assistant_id
         )
         
-        # Append the new assistant message to the conversation history
-        response_text = chat_completion.choices[0].message.content
-        conversation_history.append({"role": "assistant", "content": response_text})
-
-        # Format the history for display
-        history_formatted = "\n".join(
-            f"{m['role'].title()}: {m['content']}" for m in conversation_history
+        # Retrieve the messages to get the assistant's latest response
+        messages = client.beta.threads.messages.list(
+            thread_id=thread_id
         )
         
-        return history_formatted, conversation_history  # Return the formatted history and the raw history
+        # Extract the conversation history and the assistant's response
+        conversation_history = []
+        for msg in messages.data:
+            conversation_history.append(f"{msg['role'].title()}: {msg['content']}")
+        
+        # Join the conversation history into a formatted string
+        history_formatted = "\n".join(conversation_history)
+        
+        return history_formatted, conversation_history, thread_id  # Return the formatted history, raw history, and thread_id
     except Exception as e:
         print(f"An error occurred: {e}")
         # Return the error message and the existing history so the error message doesn't get appended
-        return "Sorry, there was an error processing your request.", history 
+        return "Sorry, there was an error processing your request.", history, thread_id 
 
 chat_interface = gr.Interface(
     fn=openai_chatbot,
@@ -48,7 +65,7 @@ chat_interface = gr.Interface(
         gr.State()  # State output to pass the history back to the function
     ],
     title="Foundation Coin Assistant",
-    description="Talk to a bot powered by OpenAI GPT-3.",
+    description="Interplanetary governance, just one query away!",
     theme=theme  # Apply the custom theme here
 )