Upload folder using huggingface_hub

app2.py

@@ -0,0 +1,83 @@
+import asyncio
+from typing import List, AsyncIterator, TypedDict
+
+from langgraph.graph import StateGraph, END
+import gradio as gr
+
+
+# 1. Define chat state
+class ChatState(TypedDict):
+    messages: List[dict]
+
+
+# 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 + " ⏳"})
+    yield {"messages": messages}
+
+
+# 3. Post-process node — replace ⏳ with ✅
+async def post_process_node(state: ChatState) -> ChatState:
+    messages = state["messages"]
+    await asyncio.sleep(1.5)
+
+    if messages[-1]["role"] == "assistant" and "⏳" in messages[-1]["content"]:
+        messages[-1]["content"] = messages[-1]["content"].replace("⏳", "✅")
+
+    return {"messages": messages}
+
+
+# 4. Define the graph
+graph_builder = StateGraph(ChatState)
+graph_builder.add_node("respond", respond_node)
+graph_builder.add_node("post_process", post_process_node)
+
+graph_builder.set_entry_point("respond")
+graph_builder.add_edge("respond", "post_process")
+graph_builder.add_edge("post_process", END)
+
+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():
+            yield node_output["messages"]
+
+
+# 6. Gradio UI
+with gr.Blocks() as demo:
+    chatbot = gr.Chatbot(label="Sanatan AI", type="messages")
+    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)
+
+    # Optional: allow enter to also send message
+    submit_event = textbox.submit(user_submit, [textbox, chatbot], [textbox, chatbot])
+    submit_event.then(bot_respond_streaming, [textbox, chatbot], chatbot)
+
+
+if __name__ == "__main__":
+    demo.launch()