Update app.py

app.py

@@ -3,66 +3,54 @@ import openai
 import gradio as gr
 from monolith_theme import MonolithTheme
 
+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)
 
-assistant_id = 'asst_wsFyYTDSGAkPibNdRGBY1Sa1'  # Replace with your Assistant's ID
-
-def openai_chatbot(message, history=None, thread_id=None):
+def openai_chatbot(message, history=None):
     try:
-        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
-            )
-
-        # Run the assistant to get the response
-        run = client.beta.threads.runs.create(
-            thread_id=thread_id,
-            assistant_id=assistant_id
+        conversation_history = history or []
+        if message:  # Only append if there is a message to send
+            conversation_history.append({"role": "user", "content": message})
+        
+        chat_completion = client.chat.completions.create(
+            model="gpt-3.5-turbo",
+            messages=conversation_history,
         )
-
-        # Retrieve the messages to get the assistant's latest response
-        messages = client.beta.threads.messages.list(
-            thread_id=thread_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
         )
-
-        # 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
+        
+        return history_formatted, conversation_history  # Return the formatted history and the raw history
     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 
 
 chat_interface = gr.Interface(
     fn=openai_chatbot,
     inputs=[
-        gr.Textbox("text", label="Your Message"),
+        gr.Textbox(label="Your Message"),
         gr.State()  # State input to hold the history
     ],
     outputs=[
-        gr.Textbox("text", label="Response"),  # Textbox for the chatbot's response
-        gr.State()  # State output to pass the history back to the function
+        gr.Textbox(label="Response", lines=10),  # More lines for displaying the history
+        gr.State(hidden=True)  # State output to pass the history back to the function
     ],
     title="Foundation Coin Assistant",
-    description="Interplanetary governance, just one query away!",
-    theme=MonolithTheme()  # Apply the custom theme here
+    description="Interplanetary Governance, one query at a time.",
+    theme=theme
 )
 
 if __name__ == "__main__":
-    chat_interface.launch(debug=True)
+    chat_interface.launch(debug=True)