Upload folder using huggingface_hub

app.py

@@ -16,11 +16,11 @@ graph = generate_graph()
 
 def init():
     load_dotenv(override=True)
-
+    
     if(os.path.isdir("./chromadb-store")):
         logger.warning("database exists locally. not downloading.")
         return
-
+    
     downloader = ZipDownloader(
         service_account_json=os.getenv("GOOGLE_SERVICE_ACCOUNT_JSON")
     )
@@ -39,29 +39,6 @@ def init_session():
     return str(uuid.uuid4())
 
 
-def chat_stream(message, history, thread_id):
-    print("received chat message for thread:", thread_id)
-    config = {"configurable": {"thread_id": thread_id}}
-
-    # Start the stream
-    stream = graph.stream(
-        {"messages": [{"role": "user", "content": message}]}, config=config
-    )
-
-    # Loop through the stream and yield what's happening
-    for event in stream:
-        step = event["step"].name
-        yield f"**Step:** `{step}`\n"
-
-        if "output" in event:
-            messages = event["output"].get("messages", [])
-            if messages:
-                # Only yield the *new* message from the step
-                latest = messages[-1]
-                if hasattr(latest, "content"):
-                    yield latest.content
-
-
 def chat(message, history, thread_id):
     print("received chat message for thread:", thread_id)
     config = {"configurable": {"thread_id": thread_id}}
@@ -108,75 +85,11 @@ chatbot = gr.Chatbot(
     show_copy_all_button=True,
     type="messages",
     height=700,
-    render_markdown=True
+    render_markdown=True,
 )
 chatInterface = gr.ChatInterface(
-    title="Sanatan-AI",
-    fn=chat_stream,
-    additional_inputs=[thread_id],
-    chatbot=chatbot
+    title="Sanatan-AI", fn=chat, additional_inputs=[thread_id], chatbot=chatbot
 )
 # initializze database
 init()
-# chatInterface.launch()
-
-with gr.Blocks() as demo:
-    thread_id = gr.State(init_session)
-
-    chatbot = gr.Chatbot(
-        label="Sanatan-Bot",
-        avatar_images=("assets/avatar_user.png", "assets/adiyen_bot.png"),
-        value=intro_messages.copy(),
-        type="messages",
-        height=700,
-        show_copy_button=True,
-        show_copy_all_button=True,
-        render_markdown=True,
-    )
-    trace_output = gr.Textbox(label="LangGraph Trace", lines=10, interactive=False)
-    msg = gr.Textbox(label="Ask your question")
-    send_btn = gr.Button("Send")
-    def chat_streaming(message, chat_history, thread_id):
-        config = {"configurable": {"thread_id": thread_id}}
-        stream = graph.stream({"messages": [{"role": "user", "content": message}]}, config=config)
-
-        # Add user message first
-        chat_history = chat_history + [{"role": "user", "content": message}]
-        yield chat_history, gr.update()
-
-        trace_lines = []
-
-        for event in stream:
-            if event.get("type") == "step":
-                step = event["step"].name
-                trace_lines.append(f"➡️ Step: `{step}`")
-                yield chat_history, gr.Textbox.update(value="\n".join(trace_lines))
-
-                output = event.get("output", {})
-                messages = output.get("messages", [])
-                if messages:
-                    latest = messages[-1]
-                    if isinstance(latest, dict):
-                        content = latest.get("content", "")
-                    else:
-                        content = getattr(latest, "content", "")
-
-                    if content:
-                        # Add assistant message progressively
-                        chat_history = chat_history + [{"role": "assistant", "content": content}]
-                        yield chat_history, gr.Textbox.update(value="\n".join(trace_lines))
-
-            elif event.get("type") == "end":
-                logger.info("All Done!")
-                trace_lines.append("✅ Done.")
-                yield chat_history, gr.Textbox.update(value="\n".join(trace_lines))
-
-
-    send_btn.click(
-        fn=chat_streaming,
-        inputs=[msg, chatbot, thread_id],
-        outputs=[chatbot, trace_output],
-    )
-
-demo.queue()
-demo.launch()
+chatInterface.launch()