Upload folder using huggingface_hub

app2.py

@@ -10,21 +10,21 @@ class ChatState(TypedDict):
     messages: List[dict]
 
 
+def start_node(state: ChatState):
+    messages = state["messages"] + [{"role": "assistant", "content": "start node: Thinking ...⏳"}]
+    return {"messages": messages}
+
+
+# 2. Respond Node — stream word by word, then ⏳
 # 2. Respond Node — stream word by word, then ⏳
 async def respond_node(state: ChatState) -> AsyncIterator[ChatState]:
     messages = state["messages"]
-    user_msg = messages[-1]["content"]
-    reply = f"respond function: {user_msg}"
-    partial = ""
-
-    for word in reply.split():
-        partial += word + " "
-        await asyncio.sleep(0.3)
-        yield {"messages": messages + [{"role": "assistant", "content": partial.strip()}]}
-
-    # Add ⏳ and yield
-    final_msg = partial.strip()
-    messages.append({"role": "assistant", "content": final_msg + " ⏳"})
+    print(messages)
+
+    await asyncio.sleep(1)  # adjust timing as needed
+    thinking_msg = "respond_node: give me some more time. please"
+    await asyncio.sleep(0.2)  # adjust timing as needed
+    messages[-1]["content"] = thinking_msg + " ⏳"
     yield {"messages": messages}
 
 
@@ -34,17 +34,19 @@ async def post_process_node(state: ChatState) -> ChatState:
     await asyncio.sleep(1.5)
 
     if messages[-1]["role"] == "assistant" and "⏳" in messages[-1]["content"]:
-        messages[-1]["content"] = messages[-1]["content"].replace("⏳", "✅")
+        messages[-1]["content"] = "post_process_node: here is your final result!" + "✅"
 
     return {"messages": messages}
 
 
 # 4. Define the graph
 graph_builder = StateGraph(ChatState)
+graph_builder.add_node("start", start_node)
 graph_builder.add_node("respond", respond_node)
 graph_builder.add_node("post_process", post_process_node)
 
-graph_builder.set_entry_point("respond")
+graph_builder.set_entry_point("start")
+graph_builder.add_edge("start", "respond")
 graph_builder.add_edge("respond", "post_process")
 graph_builder.add_edge("post_process", END)
 
@@ -54,7 +56,7 @@ graph = graph_builder.compile()
 # 5. Gradio streaming handler
 async def bot_respond_streaming(message: str, history: List[dict]) -> AsyncIterator:
     state = {"messages": (history or []) + [{"role": "user", "content": message}]}
-    
+
     async for step in graph.astream(state):
         # LangGraph yields steps like {"respond": ChatState, "post_process": ChatState}
         for node_output in step.values():
@@ -67,10 +69,10 @@ with gr.Blocks() as demo:
     with gr.Row():
         textbox = gr.Textbox(placeholder="Ask something...", scale=8, container=False)
         send_btn = gr.Button("Send", scale=1)
-    
+
     def user_submit(message, history):
         return "", history + [{"role": "user", "content": message}]
-    
+
     send_event = send_btn.click(user_submit, [textbox, chatbot], [textbox, chatbot])
     send_event.then(bot_respond_streaming, [textbox, chatbot], chatbot)