updated README

main.py

@@ -51,13 +51,18 @@ 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,
-) -> Tuple[ChatOpenAI, str, ChatHistory, List[BaseMessage]]:
+) :
     if chat is None:
         # in the queue we will store our streamed tokens
         queue = Queue()
@@ -111,7 +116,7 @@ def on_clear_click() -> Tuple[str, List, List]:
 
 def on_done_click(
     chatbot_messages, patient, 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"]