Update app.py

app.py

@@ -2,19 +2,16 @@ import os
 import openai
 import gradio as gr
 import time
-import pandas as pd
-import json
 from jtar import jtar_txt
 
 # Set up password
 username = os.getenv('username')
 password = os.getenv('password')
 
-openai.api_key = os.getenv('OPENAI_API_KEY')
-openai_api_key = openai.api_key
-
 # Set up OpenAI client
-client = openai.Client(api_key=openai_api_key)
+openai_api_key = os.getenv('openai_api_key')
+openai.api_key = openai_api_key
+client = openai.Client(api_key=openai.api_key)
 
 # Set up assistant
 assistant_id = os.getenv('assistant_id')
@@ -24,27 +21,20 @@ thread = client.beta.threads.create()
 # Get avatar link
 avatar_link = os.getenv('avatar_link')
 
-# Load and prepare the DataFrame
-df = pd.read_json('/content/IB May 2024 Results for Assistant.txt')
-df_json = df.to_json(orient='records')
-
 def chat_with_assistant(message, history):
-    # Combine the user's message with the DataFrame
-    combined_message = f"{message}\n\nHere's the DataFrame in JSON format:\n{df_json}"
-    
-    # Add the combined message to the thread
+    # Add the user's message to the thread
     client.beta.threads.messages.create(
         thread_id=thread.id,
         role="user",
-        content=combined_message
+        content=message
     )
-
+    
     # Run the assistant
     run = client.beta.threads.runs.create(
         thread_id=thread.id,
         assistant_id=assistant_id
     )
-
+    
     # Wait for the assistant's response
     while True:
         run_status = client.beta.threads.runs.retrieve(thread_id=thread.id, run_id=run.id)
@@ -54,7 +44,7 @@ def chat_with_assistant(message, history):
             assistant_response = messages.data[0].content[0].text.value
             break
         time.sleep(1)
-
+    
     return assistant_response
 
 # Custom CSS for chat bubbles and colors
@@ -65,8 +55,8 @@ custom_css = """
 
 # Create the Gradio interface
 with gr.Blocks(css=custom_css) as demo:
-    gr.HTML(f"""<img src="{avatar_link}" width="320" height="240">""")
-    gr.Markdown("# **IB Results Analyst**📊")
+    gr.HTML(f"""<img src = "https://i.postimg.cc/43kfn8Xx/a-charming-pixar-style-3d-render-of-a-dolphin-in-a-j-GGid5px-R86-Wo-FLn-Re27d-Q-w9-Mz-RVy8-Qti-PQ2-Irspua-LQ.png" width="320" height="240">""")
+    gr.Markdown("# **IB Results Analyst** 📊")
     chatbot = gr.Chatbot(
         [],
         elem_id="chatbot",
@@ -90,7 +80,6 @@ with gr.Blocks(css=custom_css) as demo:
     msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
         bot, chatbot, chatbot
     )
-
     gr.Markdown(jtar_txt)
 
-demo.launch(auth=(username, password))
+demo.launch(auth=(username, password))"