Upload folder using huggingface_hub

chatgpt.py

@@ -12,8 +12,8 @@ from interview_protocol import protocols as interview_protocols
 
 model = whisper.load_model("base")
 
-base_url = "https://llm4socialisolation-fd4082d0a518.herokuapp.com"
-# base_url = "http://localhost:8080"
+# base_url = "https://llm4socialisolation-fd4082d0a518.herokuapp.com"
+base_url = "http://localhost:8080"
 
 timeout = 60
 
@@ -82,10 +82,13 @@ async def get_backend_response(api_key, patient_prompt, username, chatbot_type):
         except Exception as e:
             return f"Error contacting backend service: {str(e)}"
 
-async def save_conversation_and_memory(username):
+async def save_conversation_and_memory(username, chatbot_type):
     url = f"{base_url}/save/end_and_save"
     headers = {'Content-Type': 'application/json'}
-    data = {'username': username}
+    data = {
+        'username': username,
+        'chatbot_type': chatbot_type
+    }
     async with httpx.AsyncClient(timeout=timeout) as client:
         try:
             response = await client.post(url, json=data, headers=headers)
@@ -97,10 +100,13 @@ async def save_conversation_and_memory(username):
         except Exception as e:
             return f"Error contacting backend service: {str(e)}"
 
-async def get_conversation_histories(username):
+async def get_conversation_histories(username, chatbot_type):
     url = f"{base_url}/save/download_conversations"
     headers = {'Content-Type': 'application/json'}
-    data = {'username': username}
+    data = {
+        'username': username,
+        'chatbot_type': chatbot_type
+    }
     async with httpx.AsyncClient(timeout=timeout) as client:
         try:
             response = await client.post(url, json=data, headers=headers)
@@ -112,8 +118,8 @@ async def get_conversation_histories(username):
         except Exception as e:
             return []
 
-def download_conversations(username):
-    conversation_histories = asyncio.run(get_conversation_histories(username))
+def download_conversations(username, chatbot_type):
+    conversation_histories = asyncio.run(get_conversation_histories(username, chatbot_type))
     files = []
     temp_dir = tempfile.mkdtemp()
     for conversation_entry in conversation_histories:
@@ -132,10 +138,13 @@ def download_conversations(username):
         files.append(temp_file_path)
     return files
 
-async def get_biography(username):
+async def get_biography(username, chatbot_type):
     url = f"{base_url}/save/generate_autobiography"
     headers = {'Content-Type': 'application/json'}
-    data = {'username': username}
+    data = {
+        'username': username,
+        'chatbot_type': chatbot_type
+    }
     async with httpx.AsyncClient(timeout=timeout) as client:
         try:
             response = await client.post(url, json=data, headers=headers)
@@ -148,8 +157,8 @@ async def get_biography(username):
         except Exception as e:
             return f"Error contacting backend service: {str(e)}"
 
-def download_biography(username):
-    biography_text = asyncio.run(get_biography(username))
+def download_biography(username, chatbot_type):
+    biography_text = asyncio.run(get_biography(username, chatbot_type))
     if not biography_text or "Failed" in biography_text or "Error" in biography_text:
         return gr.update(value=None, visible=False), biography_text
     temp_dir = tempfile.mkdtemp()
@@ -187,8 +196,8 @@ def set_initialize_button(api_key_input, chapter_name, topic_name, username_inpu
     print(message)
     return message, api_key_input
 
-def save_conversation(username):
-    response = asyncio.run(save_conversation_and_memory(username))
+def save_conversation(username, chatbot_type):
+    response = asyncio.run(save_conversation_and_memory(username, chatbot_type))
     return response
 
 def start_recording(audio_file):
@@ -390,19 +399,19 @@ with gr.Blocks() as app:
 
             download_button.click(
                 fn=download_conversations,
-                inputs=[username_input],
+                inputs=[username_input, chatbot_type_state],
                 outputs=gr.Files()
             )
 
             download_biography_button.click(
                 fn=download_biography,
-                inputs=[username_input],
+                inputs=[username_input, chatbot_type_state],
                 outputs=[gr.File(label="Biography.txt"), gr.Textbox(visible=False)]
             )
 
             save_conversation_button.click(
                 fn=save_conversation,
-                inputs=[username_input],
+                inputs=[username_input, chatbot_type_state],
                 outputs=None
             )