Update app.py

app.py

@@ -33,7 +33,7 @@ default_tools = [{"type": "function", "function": {"name": "get_current_weather"
                 }
             }}]
 
-def apply_chat_template(model_name, test_conversation, add_generation_prompt, cleanup_whitespace, hf_token, kwargs):
+def apply_chat_template(model_name, test_conversation, add_generation_prompt, cleanup_whitespace, hf_token, tools):
     try:
         if hf_token:
             login(token=hf_token)  # Ensure login is successful
@@ -49,7 +49,7 @@ def apply_chat_template(model_name, test_conversation, add_generation_prompt, cl
             conversation,
             tokenize=False,
             add_generation_prompt=add_generation_prompt,
-            tools=default_tools
+            tools=tools
         )
         return formatted
     except Exception as e:
@@ -70,6 +70,9 @@ with gr.Blocks() as demo:
         with gr.Column():
             output = gr.TextArea(label="Formatted Conversation", interactive=False, lines=12)
 
+    # Use gr.State() to pass default_tools correctly
+    tools_state = gr.State(default_tools)
+
     format_button.click(
         fn=apply_chat_template,
         inputs=[
@@ -78,9 +81,9 @@ with gr.Blocks() as demo:
             add_generation_prompt_checkbox,
             cleanup_whitespace_checkbox,
             hf_token_input,
-            default_tools  # Default tools parameter
+            tools_state  # Wrapped inside gr.State()
         ],
         outputs=output
     )
 
-demo.launch()
+demo.launch()