bugfix for missing chat

app.py

@@ -13,7 +13,7 @@ def delete_db(req: gr.Request):
     dir_path = TEMP_DIR / str(req.session_hash)
     if os.path.exists(dir_path):
         shutil.rmtree(dir_path)
-        message_dict[req.session_hash] = None
+        message_dict[req.session_hash] = {}
 
 if "OPENAI_API_KEY" not in os.environ:
     os.environ["OPENAI_API_KEY"] = getpass("Enter OpenAI API key:")

templates/data_file.py

@@ -68,7 +68,8 @@ with gr.Blocks() as demo:
     @gr.render(inputs=file_output)
     def data_options(filename, request: gr.Request):
         print(filename)
-        message_dict[request.session_hash] = {}
+        if request.session_hash not in message_dict:
+            message_dict[request.session_hash] = {}
         message_dict[request.session_hash]['file_upload'] = None
         if filename:
             process_message = process_upload(filename, request.session_hash)

templates/sql_db.py

@@ -36,10 +36,11 @@ with gr.Blocks() as demo:
 
     @gr.render(inputs=[sql_url,sql_port,sql_user,sql_pass,sql_db_name], triggers=[submit.click])
     def sql_chat(request: gr.Request, url=sql_url.value, sql_port=sql_port.value, sql_user=sql_user.value, sql_pass=sql_pass.value, sql_db_name=sql_db_name.value):
+        if request.session_hash not in message_dict:
+            message_dict[request.session_hash] = {}
         message_dict[request.session_hash]['sql'] = None
         if url:
             print("SQL APP")
-            print(request)
             process_message = process_sql_db(url, sql_user, sql_port, sql_pass, sql_db_name, request.session_hash)
             gr.HTML(value=process_message[1], padding=False)
             if process_message[0] == "success":