Upload folder using huggingface_hub

app.py

@@ -95,17 +95,18 @@ thinking_verbs = [
 
 async def chat_wrapper(message, history, thread_id, debug):
     if debug:
-        async for chunk in chat_streaming(message, history, thread_id):
+        async for chunk in chat_streaming(debug, message, history, thread_id):
             yield chunk
     else:
-        response = chat(message, history, thread_id)
+        response = chat(debug, message, history, thread_id)
         yield response
 
 
-def chat(message, history, thread_id):
+def chat(debug_mode, message, history, thread_id):
     config = {"configurable": {"thread_id": thread_id}}
     response = graph.invoke(
-        {"messages": [{"role": "user", "content": message}]}, config=config
+        {"debug_mode": debug_mode, "messages": [{"role": "user", "content": message}]},
+        config=config,
     )
     return response["messages"][-1].content
 
@@ -139,8 +140,11 @@ def get_args_for_toolcall(tool_calls_buffer: dict, tool_call_id: str):
     )
 
 
-async def chat_streaming(message, history, thread_id):
-    state = {"messages": (history or []) + [{"role": "user", "content": message}]}
+async def chat_streaming(debug_mode: bool, message, history, thread_id):
+    state = {
+        "debug_mode": debug_mode,
+        "messages": (history or []) + [{"role": "user", "content": message}],
+    }
     config = {"configurable": {"thread_id": thread_id}}
     start_time = time.time()
     streamed_response = ""
@@ -330,31 +334,11 @@ supported_scriptures = "\n - ".join(
     ]
 )
 
-intro_messages = [
-    {
-        "role": "assistant",
-        "content": f"Namaskaram 🙏 I am Sanatan-Bot and I can help you explore the following scriptures:\n\n - {supported_scriptures}",
-    },
-    {
-        "role": "assistant",
-        "content": """
-#### Example questions you can ask:
-  - How is the form of Vishnu described across the scriptures?
-  - What teachings did Krishna share in the Gita?
-  - How did Arjuna respond upon witnessing Krishna’s Vishwarupa?
-  - What are some names of Vishnu from the Sahasranamam?
-  - Give me a pasuram by Andal
-  - explain sakrudeva prapannaaya shlokam in ramayana
-  - give the shlokam in ramayanam that vibheeshana uses to perform sharanagathi to rama, give the sanskrit shlokam and its meaning
-  - give one name in vishnu sahasranamam related to "cause"
-  - explain name of vishnu in sahasranamam related to wheel
-    """,
-    },
-]
-
 init()
 
-message_textbox = gr.Textbox(placeholder="Search the scriptures ...", submit_btn=True, stop_btn=True)
+message_textbox = gr.Textbox(
+    placeholder="Search the scriptures ...", submit_btn=True, stop_btn=True
+)
 
 with gr.Blocks(
     theme=gr.themes.Citrus(),
@@ -424,7 +408,10 @@ with gr.Blocks(
     }
     """,
 ) as app:
-    with gr.Sidebar() as sidebar:
+    show_sidebar = gr.State(True)
+
+    # with gr.Column(scale=1, visible=show_sidebar.value) as sidebar_container:
+    with gr.Sidebar(open=show_sidebar.value) as sidebar:
         # session_id = gr.Textbox(value=f"Thread: {thread_id}")
         # gr.Markdown(value=f"{'\n'.join([msg['content'] for msg in intro_messages])}")
         gr.Markdown(
@@ -439,6 +426,10 @@ with gr.Blocks(
                 await asyncio.sleep(0.05)
             return
 
+        def close_side_bar():
+            print("close_side_bar invoked")
+            yield gr.update(open=False)
+
         for scripture in sorted(SanatanConfig.scriptures, key=lambda d: d.get("title")):
             with gr.Accordion(label=f"{scripture['title']}", open=False):
                 gr.Markdown(f"* Source: [🔗 click here]({scripture['source']})")
@@ -449,7 +440,7 @@ with gr.Blocks(
                         scripture["example_labels"], scripture["examples"]
                     ):
                         btn = gr.Button(value=f"{example_label}", size="sm")
-                        btn.click(
+                        btn.click(close_side_bar,outputs=[sidebar]).then(
                             populate_chat_input,
                             inputs=[gr.State(example_text)],
                             outputs=[message_textbox],

app3.py

@@ -0,0 +1,15 @@
+import gradio as gr
+
+def collapse_sidebar():
+    # return a config change for the Sidebar component
+    return gr.update(open=False)
+
+with gr.Blocks() as demo:
+    with gr.Sidebar() as sidebar:
+        gr.Markdown("Sidebar content")
+        close_btn = gr.Button("Close sidebar")
+
+    # clicking the button tells Gradio to update the Sidebar's 'open' prop
+    close_btn.click(fn=collapse_sidebar, inputs=None, outputs=sidebar)
+
+demo.launch()

graph_helper.py

@@ -25,15 +25,21 @@ logger.setLevel(logging.INFO)
 
 
 class ChatState(TypedDict):
+    debug_mode : bool = True
     messages: Annotated[list[str], add_messages]
 
+def check_debug_condition(state: ChatState) -> str:
+    if state["debug_mode"]:
+        return "validator"
+    else:
+        return "__end__"
 
 def branching_condition(state: ChatState) -> str:
     last_message = state["messages"][-1]
     if hasattr(last_message, "tool_calls") and last_message.tool_calls:
         return "tools"
     else:
-        return "validator"
+        return check_debug_condition(state)
 
 
 def generate_graph() -> CompiledStateGraph:
@@ -220,6 +226,7 @@ Where:
         {
             "tools": "tools",
             "validator": "validator",
+            "__end__" : END
         },
     )
     graph.add_edge("tools", "llm")