Upload folder using huggingface_hub

app.py

@@ -5,6 +5,8 @@ from utils import stream_from_transformers_agent
 from prompts import SQUAD_REACT_CODE_SYSTEM_PROMPT
 from tools.squad_tools import SquadRetrieverTool, SquadQueryTool
 from tools.text_to_image import TextToImageTool
+from gradio.context import Context
+from gradio import Request
 from dotenv import load_dotenv
 
 load_dotenv()
@@ -44,15 +46,38 @@ def add_message(message, messages):
     messages.append(ChatMessage(role="user", content=message))
     return messages
 
-def interact_with_agent(messages):
+sessions = {}
+
+def interact_with_agent(messages, request: Request):
     prompt = messages[-1]['content']
+    agent.logs = sessions.get(request.session_hash + "_logs", [])
     for msg in stream_from_transformers_agent(agent, prompt):
         messages.append(msg)
         yield messages
     yield messages
 
+def persist(component):
+
+    def resume_session(value, request: Request):
+        print(f"Resuming session for {request.session_hash}")
+        state = sessions.get(request.session_hash, value)
+        agent.logs = sessions.get(request.session_hash + "_logs", [])
+        return state
+
+    def update_session(value, request: Request):
+        print(f"Updating persisted session state for {request.session_hash}")
+        sessions[request.session_hash] = value
+        sessions[request.session_hash + "_logs"] = agent.logs
+        return
+
+    Context.root_block.load(resume_session, inputs=[component], outputs=component)
+    component.change(update_session, inputs=[component], outputs=[])
+
+    return component
+
 with gr.Blocks(fill_height=True) as demo:
-    chatbot = gr.Chatbot(
+    chatbot = persist(gr.Chatbot(
+        value=[],
         label="SQuAD Agent",
         type="messages",
         avatar_images=(
@@ -60,7 +85,7 @@ with gr.Blocks(fill_height=True) as demo:
             "https://em-content.zobj.net/source/twitter/53/robot-face_1f916.png",
         ),
         scale=1,
-        bubble_full_width=False,
+        # bubble_full_width=False,
         autoscroll=True,
         show_copy_all_button=True,
         show_copy_button=True,
@@ -76,7 +101,7 @@ with gr.Blocks(fill_height=True) as demo:
                 "text": "Draw a picture of whatever is on top of the Notre Dame building.",
             },
         ],
-    )
+    ))
     text_input = gr.Textbox(lines=1, label="Chat Message", scale=0)
     chat_msg = text_input.submit(add_message, [text_input, chatbot], [chatbot])
     bot_msg = chat_msg.then(interact_with_agent, [chatbot], [chatbot])