clear button bug

app.py

@@ -25,7 +25,7 @@ def create_ui():
 
         # Load the buttons for the interface
         load_buttons(query, model,
-                     sustainable, starting_point,
+                     sustainable, country, starting_point,
                      max_new_tokens, temperature,
                      output,
                      generate_text_fn=generate_text,

src/ui/components/actions.py

@@ -4,7 +4,7 @@ from src.pipeline import pipeline
 
 
 def clear():
-    return None, None, None, None, None
+    return None, None, None, None, None, None
 
 
 def generate_text(query_text: str,

src/ui/components/static.py

@@ -51,6 +51,7 @@ def load_buttons(
         query: gr.components.Component,
         model: gr.components.Component,
         sustainable: gr.components.Component,
+        country: gr.components.Component,
         starting_point: gr.components.Component,
         max_new_tokens: gr.components.Component,
         temperature: gr.components.Component,
@@ -92,12 +93,12 @@ def load_buttons(
         clear_btn.click(
             fn=clear_fn,  # Function to clear inputs
             inputs=[],  # No inputs for clearing
-            outputs=[query, model, sustainable, starting_point, output]  # Clear all inputs and output
+            outputs=[query, model, sustainable, starting_point, country, output]  # Clear all inputs and output
         )
         cancel_btn.click(
             fn=clear_fn,  # Function to cancel and clear inputs
             inputs=[],  # No inputs for cancel
-            outputs=[query, model, sustainable, starting_point, output]  # Clear all inputs and output
+            outputs=[query, model, sustainable, starting_point, country, output]  # Clear all inputs and output
         )
     return btns