Update app.py

app.py

@@ -6,25 +6,51 @@ import uuid
 import time
 import os
 import openai
+import pandas as pd
 
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 
 # Set your OpenAI Assistant ID here
 assistant_id = os.getenv("assistant_id")
 
+
+# Step 1: Clone the Dataset Repository
+repo = Repository(
+    local_dir="Test_csv",  # Local directory to clone the repository
+    repo_type="dataset",  
+    clone_from="Anne31415/Test_csv", 
+    token=os.environ["HUB_TOKEN"]  # Token for authentication
+)
+repo.git_pull()  # Pull the latest changes (if any)
+
+# Step 2: Load the PDF File
+file_path = "Test_csv/Basic_Info_GrunddatenKH141123.csv" 
+
+file = client.files.create(
+  file=open(file_path, "rb"),
+  purpose='assistants'
+)
+
+
+
 def main():
 
     if 'client' not in st.session_state:
         # Initialize the client
         st.session_state.client = openai.OpenAI()
 
+        st.session_state.file = st.session_state.client.files.create(
+            file,
+            purpose='assistants'
+        )
 
         # Step 1: Create an Assistant
         st.session_state.assistant = st.session_state.client.beta.assistants.create(
-            name="Customer Service Assistant",
-            instructions="You are a customer support chatbot. Use your knowledge base to best respond to customer queries.",
+            name="Medical Data Analyst",
+            instructions="You are a data analyst for medical data. Use your csv knowlagebase to answer the given questions. Make plots if it fits the question.",
             model="gpt-4-1106-preview",
-            tools=[{"type": "retrieval"}]
+            file_ids=[st.session_state.file.id],
+            tools=[{"type": "code_interpreter"}]
         )
 
         # Step 2: Create a Thread