new ui elements

Gradio_UI.py

@@ -274,6 +274,9 @@ class GradioUI:
             "",
         )
 
+    def set_agent_steps(self, steps):
+        self.agent.max_steps = steps
+
     def launch(self, **kwargs):
         import gradio as gr
 
@@ -333,6 +336,9 @@ class GradioUI:
                     [upload_file, file_uploads_log],
                     [upload_status, file_uploads_log],
                 )
+            with gr.Row():
+                steps_input = gr.Slider(0, 12, value=4, step=1, label="Max. Number of Steps")
+                reset = gr.Button(value="Reset Agent")
             text_input = gr.Textbox(lines=1, label="Chat Message")
             text_input.submit(
                 self.log_user_message,
@@ -343,6 +349,7 @@ class GradioUI:
                 examples=[["Tell me a joke based on the current local time"],["Given the current local time, what is a fun activity to do?"],["When asked for the current local time, add 6 hours to it. What is the current local time?"], ["Find significant events that happend exactly one year ago"], ["Generate a bold picture inspired by the current local time"]],
             inputs=[text_input],
             )
+        
         demo.launch(debug=True, share=True, ssr_mode=False, allowed_paths=["Cyberpunk.otf"], **kwargs)
 
 __all__ = ["stream_to_gradio", "GradioUI"]

app.py

@@ -7,7 +7,6 @@ from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
-
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -23,33 +22,36 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
-final_answer = FinalAnswerTool() 
+final_answer = FinalAnswerTool()
 
 model = HfApiModel(
-max_tokens=2096,
-temperature=0.5,
-model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
-custom_role_conversions=None,
+    max_tokens=2096,
+    temperature=0.5,
+    model_id="Qwen/Qwen2.5-Coder-32B-Instruct",  # it is possible that this model may be overloaded
+    custom_role_conversions=None,
 )
 
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
-with open("prompts.yaml", 'r') as stream:
+with open("prompts.yaml", "r") as stream:
     prompt_templates = yaml.safe_load(stream)
-    
+
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_in_timezone, image_generation_tool, DuckDuckGoSearchTool()], ## add your tools here (don't remove final answer)
+    tools=[
+        final_answer,
+        get_current_time_in_timezone,
+        image_generation_tool,
+        DuckDuckGoSearchTool(),
+    ],  ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,
     planning_interval=None,
     name=None,
     description=None,
-    prompt_templates=prompt_templates
+    prompt_templates=prompt_templates,
 )
 
-print(f"docstring {get_current_time_in_timezone.__doc__}")
-print(f"string {get_current_time_in_timezone}")
 GradioUI(agent).launch()