updated README

README.md

@@ -4,7 +4,7 @@ emoji: "🧪"
 colorFrom: "#009FFD"
 colorTo: "#2A2A72"
 sdk: "gradio"
-sdk_version: "3.31.0"
+sdk_version: "3.32.0"
 python_version: "3.9"
 app_file: main.py
 pinned: false

main.py

@@ -51,18 +51,13 @@ with open("data/patients.json") as f:
 
 patients_names = [el["name"] for el in patiens]
 
-# it was 
-# chat: ChatOpenAI
-# message: str
-# chatbot_messages: ChatHistory
-# messages: List[BaseMessage]
-# -> -> Tuple[ChatOpenAI, str, ChatHistory, List[BaseMessage]]
+
 def message_handler(
-    chat,
-    message,
-    chatbot_messages,
-    messages,
-) :
+    chat: Optional[ChatOpenAI],
+    message: str,
+    chatbot_messages: ChatHistory,
+    messages: List[BaseMessage],
+) -> Tuple[ChatOpenAI, str, ChatHistory, List[BaseMessage]]:
     if chat is None:
         # in the queue we will store our streamed tokens
         queue = Queue()
@@ -115,8 +110,8 @@ def on_clear_click() -> Tuple[str, List, List]:
 
 
 def on_done_click(
-    chatbot_messages, patient, user
-):
+    chatbot_messages: ChatHistory, patient: str, user: User
+) -> Tuple[str, List, List]:
     logging.info(f"Saving chat for user={user}")
     add_chat_by_uid(
         client, Chat(patient=patient, messages=chatbot_messages), user["uid"]
@@ -124,7 +119,7 @@ def on_done_click(
     return on_clear_click()
 
 
-def on_apply_settings_click(model_name, temperature):
+def on_apply_settings_click(model_name: str, temperature: float):
     logging.info(
         f"Applying settings: model_name={model_name}, temperature={temperature}"
     )
@@ -148,7 +143,7 @@ def on_drop_down_change(selected_item, messages):
 
 
 def on_demo_load(url_params):
-    username = url_params["username"]
+    username = url_params.get("username", "test")
     logging.info(f"Getting user for username={username}")
     create_user(client, User(username=username, uid=None))
     user = get_user_by_username(client, username)

requirements.txt

@@ -1,4 +1,4 @@
-gradio==3.31.0
+gradio==3.32.0
 openai
 elevenlabs
 python-dotenv