Update app.py

app.py

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