Update app.py

app.py

@@ -3,69 +3,66 @@ import openai
 import gradio as gr
 from monolith_theme import MonolithTheme
 
-# Set your API key from an environment variable or directly
-openai.api_key = os.getenv("OPENAI_API_KEY")
+# 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)
 
-theme = MonolithTheme()
-
-# Initialize the assistant ID you created via the OpenAI platform
-assistant_id = 'asst_wsFyYTDSGAkPibNdRGBY1Sa1'  # Replace with your Assistant ID
+assistant_id = 'asst_wsFyYTDSGAkPibNdRGBY1Sa1'  # Replace with your Assistant's ID
 
 def openai_chatbot(message, history=None, thread_id=None):
     try:
-        # If there's no thread_id, create a new thread (new conversation)
         if not thread_id:
-            thread = openai.Thread.create(assistant=assistant_id)
-            thread_id = thread['id']
-        
-        # Add the user's message to the thread
+            # Create a new thread if one does not exist
+            thread = client.beta.threads.create()
+            thread_id = thread.id
+
         if message:
-            openai.Message.create(
-                thread=thread_id,
+            # Add the user's message to the thread
+            client.beta.threads.messages.create(
+                thread_id=thread_id,
                 role="user",
                 content=message
             )
-        
-        # Run the assistant on the thread to generate a response
-        run = openai.Run.create(
-            assistant=assistant_id,
-            thread=thread_id
+
+        # Run the assistant to get the response
+        run = client.beta.threads.runs.create(
+            thread_id=thread_id,
+            assistant_id=assistant_id
         )
-        
-        # Retrieve the messages from the thread
-        messages = openai.Message.list(
-            thread=thread_id
+
+        # Retrieve the messages to get the assistant's latest response
+        messages = client.beta.threads.messages.list(
+            thread_id=thread_id
         )
-        
-        # Format the conversation history
-        history_formatted = [f"{msg['role'].capitalize()}: {msg['content']}" for msg in messages['data']]
-        
-        # The bot's latest response will be the last message in the list
-        bot_response = history_formatted[-1] if history_formatted else "No response from the bot."
-        
-        return bot_response, history_formatted, thread_id
+
+        # 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, thread_id
 
-# Create Gradio interface
-iface = gr.Interface(
+chat_interface = gr.Interface(
     fn=openai_chatbot,
     inputs=[
-        gr.Textbox(label="Your Message"),
-        gr.State(label="Conversation History"),
-        gr.State(label="Thread ID")
+        gr.Textbox("text", label="Your Message"),
+        gr.State()  # State input to hold the history
     ],
     outputs=[
-        gr.Textbox(label="Chatbot's Reply"),
-        gr.State(label="Updated History"),
-        gr.State(label="Updated Thread ID")
+        gr.Textbox("text", label="Response"),  # Textbox for the chatbot's response
+        gr.State()  # State output to pass the history back to the function
     ],
-    title="OpenAI Chatbot",
-    description="Talk to a bot powered by OpenAI GPT-3.",
-    theme=theme
+    title="Foundation Coin Assistant",
+    description="Interplanetary governance, just one query away!",
+    theme=MonolithTheme()  # Apply the custom theme here
 )
 
-# Launch the Gradio app
 if __name__ == "__main__":
-    iface.launch()
+    chat_interface.launch(debug=True)