camera

Gradio_UI.py

@@ -22,11 +22,17 @@ from typing import Optional
 import datetime
 import pytz
 
-from smolagents.agent_types import AgentAudio, AgentImage, AgentText, handle_agent_output_types
+from smolagents.agent_types import (
+    AgentAudio,
+    AgentImage,
+    AgentText,
+    handle_agent_output_types,
+)
 from smolagents.agents import ActionStep, MultiStepAgent
 from smolagents.memory import MemoryStep
 from smolagents.utils import _is_package_available
 
+
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
     Args:
@@ -41,6 +47,7 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+
 def pull_messages_from_step(
     step_log: MemoryStep,
 ):
@@ -49,7 +56,9 @@ def pull_messages_from_step(
 
     if isinstance(step_log, ActionStep):
         # Output the step number
-        step_number = f"Step {step_log.step_number}" if step_log.step_number is not None else ""
+        step_number = (
+            f"Step {step_log.step_number}" if step_log.step_number is not None else ""
+        )
         yield gr.ChatMessage(role="assistant", content=f"**{step_number}**")
 
         # First yield the thought/reasoning from the LLM
@@ -57,9 +66,15 @@ def pull_messages_from_step(
             # Clean up the LLM output
             model_output = step_log.model_output.strip()
             # Remove any trailing <end_code> and extra backticks, handling multiple possible formats
-            model_output = re.sub(r"```\s*<end_code>", "```", model_output)  # handles ```<end_code>
-            model_output = re.sub(r"<end_code>\s*```", "```", model_output)  # handles <end_code>```
-            model_output = re.sub(r"```\s*\n\s*<end_code>", "```", model_output)  # handles ```\n<end_code>
+            model_output = re.sub(
+                r"```\s*<end_code>", "```", model_output
+            )  # handles ```<end_code>
+            model_output = re.sub(
+                r"<end_code>\s*```", "```", model_output
+            )  # handles <end_code>```
+            model_output = re.sub(
+                r"```\s*\n\s*<end_code>", "```", model_output
+            )  # handles ```\n<end_code>
             model_output = model_output.strip()
             yield gr.ChatMessage(role="assistant", content=model_output)
 
@@ -79,8 +94,12 @@ def pull_messages_from_step(
 
             if used_code:
                 # Clean up the content by removing any end code tags
-                content = re.sub(r"```.*?\n", "", content)  # Remove existing code blocks
-                content = re.sub(r"\s*<end_code>\s*", "", content)  # Remove end_code tags
+                content = re.sub(
+                    r"```.*?\n", "", content
+                )  # Remove existing code blocks
+                content = re.sub(
+                    r"\s*<end_code>\s*", "", content
+                )  # Remove end_code tags
                 content = content.strip()
                 if not content.startswith("```python"):
                     content = f"```python\n{content}\n```"
@@ -106,7 +125,11 @@ def pull_messages_from_step(
                     yield gr.ChatMessage(
                         role="assistant",
                         content=f"{log_content}",
-                        metadata={"title": "📝 Execution Logs", "parent_id": parent_id, "status": "done"},
+                        metadata={
+                            "title": "📝 Execution Logs",
+                            "parent_id": parent_id,
+                            "status": "done",
+                        },
                     )
 
             # Nesting any errors under the tool call
@@ -114,7 +137,11 @@ def pull_messages_from_step(
                 yield gr.ChatMessage(
                     role="assistant",
                     content=str(step_log.error),
-                    metadata={"title": "💥 Error", "parent_id": parent_id, "status": "done"},
+                    metadata={
+                        "title": "💥 Error",
+                        "parent_id": parent_id,
+                        "status": "done",
+                    },
                 )
 
             # Update parent message metadata to done status without yielding a new message
@@ -122,17 +149,25 @@ def pull_messages_from_step(
 
         # Handle standalone errors but not from tool calls
         elif hasattr(step_log, "error") and step_log.error is not None:
-            yield gr.ChatMessage(role="assistant", content=str(step_log.error), metadata={"title": "💥 Error"})
+            yield gr.ChatMessage(
+                role="assistant",
+                content=str(step_log.error),
+                metadata={"title": "💥 Error"},
+            )
 
         # Calculate duration and token information
         step_footnote = f"{step_number}"
-        if hasattr(step_log, "input_token_count") and hasattr(step_log, "output_token_count"):
-            token_str = (
-                f" | Input-tokens:{step_log.input_token_count:,} | Output-tokens:{step_log.output_token_count:,}"
-            )
+        if hasattr(step_log, "input_token_count") and hasattr(
+            step_log, "output_token_count"
+        ):
+            token_str = f" | Input-tokens:{step_log.input_token_count:,} | Output-tokens:{step_log.output_token_count:,}"
             step_footnote += token_str
         if hasattr(step_log, "duration"):
-            step_duration = f" | Duration: {round(float(step_log.duration), 2)}" if step_log.duration else None
+            step_duration = (
+                f" | Duration: {round(float(step_log.duration), 2)}"
+                if step_log.duration
+                else None
+            )
             step_footnote += step_duration
         step_footnote = f"""<span style="color: #bbbbc2; font-size: 12px;">{step_footnote}</span> """
         yield gr.ChatMessage(role="assistant", content=f"{step_footnote}")
@@ -155,7 +190,9 @@ def stream_to_gradio(
     total_input_tokens = 0
     total_output_tokens = 0
 
-    for step_log in agent.run(task, stream=True, reset=reset_agent_memory, additional_args=additional_args):
+    for step_log in agent.run(
+        task, stream=True, reset=reset_agent_memory, additional_args=additional_args
+    ):
         # Track tokens if model provides them
         if hasattr(agent.model, "last_input_token_count"):
             total_input_tokens += agent.model.last_input_token_count
@@ -188,7 +225,9 @@ def stream_to_gradio(
             content={"path": final_answer.to_string(), "mime_type": "audio/wav"},
         )
     else:
-        yield gr.ChatMessage(role="assistant", content=f"**Final answer:** {str(final_answer)}")
+        yield gr.ChatMessage(
+            role="assistant", content=f"**Final answer:** {str(final_answer)}"
+        )
 
 
 class GradioUI:
@@ -258,10 +297,14 @@ class GradioUI:
         sanitized_name = "".join(sanitized_name)
 
         # Save the uploaded file to the specified folder
-        file_path = os.path.join(self.file_upload_folder, os.path.basename(sanitized_name))
+        file_path = os.path.join(
+            self.file_upload_folder, os.path.basename(sanitized_name)
+        )
         shutil.copy(file.name, file_path)
 
-        return gr.Textbox(f"File uploaded: {file_path}", visible=True), file_uploads_log + [file_path]
+        return gr.Textbox(
+            f"File uploaded: {file_path}", visible=True
+        ), file_uploads_log + [file_path]
 
     def log_user_message(self, text_input, file_uploads_log):
         return (
@@ -287,10 +330,13 @@ class GradioUI:
     def launch(self, **kwargs):
         import gradio as gr
 
-        with gr.Blocks(fill_height=True, theme='crcdng/cyber', css_paths=["cyberpunk.css",  "scrolling_text.css"]) as demo:
-            
-            title_html=(
-                """
+        with gr.Blocks(
+            fill_height=True,
+            theme="crcdng/cyber",
+            css_paths=["cyberpunk.css", "scrolling_text.css"],
+        ) as demo:
+
+            title_html = """
                 <style>
                 @font-face { 
                     font-family: Cyberpunk;
@@ -303,81 +349,115 @@ class GradioUI:
                 <h1 style='font-family: Cyberpunk; font-size: 42px;'> Your Cyberpunk Local Time Terminal </h1>
                 </center>
                 """
-            )
 
-            description_html=(
-                """
+            description_html = """
                 <center><p style='font-size: 24px;'> 
                 Welcome to ChronoCore-77, the bleeding-edge time terminal jacked straight into the neon veins of Night City. Whether you're dodging corpos, chasing edgerunner gigs, or just trying to sync your implant clock, I've got the local time locked and loaded. No glitches [OK a few...], no lag—just pure, precise chrono-data ripped straight from the grid. Stay sharp, choom. Time waits for no one.
                 </p></center>
                 """
-            )
 
-            banner_html=(
-                """
+            banner_html = """
                 <div class="cyber-banner-short bg-purple fg-white cyber-glitch-1">
                 <div style="margin-left: auto; width: 80%;" id="scroll-container">
                 <div id="scroll-text"> ---- Tannhäuser Gate Approved ---- Special Offer Today Only ----- Free Access Credits </div>
                 </div>
                 </div>
                 """
-            )
 
             with gr.Row():
-                title=gr.HTML(banner_html) 
+                title = gr.HTML(banner_html)
             with gr.Row():
                 timer = gr.Timer(1)
                 gr.Textbox(render=False)
                 time_display = gr.Textbox(label="Time", elem_classes="cyber-glitch-4")
                 gr.Textbox(render=False)
                 import time
-                timer.tick(lambda: get_current_time_in_timezone(time.tzname[0]), outputs=time_display)
+
+                timer.tick(
+                    lambda: get_current_time_in_timezone(time.tzname[0]),
+                    outputs=time_display,
+                )
             with gr.Row():
-                title=gr.HTML(title_html) 
+                title = gr.HTML(title_html)
             with gr.Row():
-                description=gr.HTML(description_html)     
+                description = gr.HTML(description_html)
             stored_messages = gr.State([])
             file_uploads_log = gr.State([])
             with gr.Row(equal_height=True):
                 chatbot = gr.Chatbot(
-                label="Agent",
-                type="messages",
-                avatar_images=(
-                "https://huggingface.co/spaces/crcdng/First_agent_template/resolve/main/agent_b.jpg",
-                "https://huggingface.co/spaces/crcdng/First_agent_template/resolve/main/agent_a.jpg",
-                ),
-                resizeable=True,
-                scale=2,
-                elem_classes="cyber-glitch-1",
+                    label="Agent",
+                    type="messages",
+                    avatar_images=(
+                        "https://huggingface.co/spaces/crcdng/First_agent_template/resolve/main/agent_b.jpg",
+                        "https://huggingface.co/spaces/crcdng/First_agent_template/resolve/main/agent_a.jpg",
+                    ),
+                    resizeable=True,
+                    scale=2,
+                    elem_classes="cyber-glitch-1",
+                )
+                gr.Model3D(
+                    "terminal.glb",
+                    display_mode="solid",
+                    camera_position=(90, 180, 2),
+                    label="A view of the ChronoCore-77",
                 )
-                gr.Model3D("terminal.glb", display_mode= "wireframe", label="A view of the ChronoCore-77")
             # If an upload folder is provided, enable the upload feature
             if self.file_upload_folder is not None:
                 upload_file = gr.File(label="Upload a file")
-                upload_status = gr.Textbox(label="Upload Status", interactive=False, visible=False)
+                upload_status = gr.Textbox(
+                    label="Upload Status", interactive=False, visible=False
+                )
                 upload_file.change(
                     self.upload_file,
                     [upload_file, file_uploads_log],
                     [upload_status, file_uploads_log],
                 )
             with gr.Row(equal_height=True):
-                steps_input = gr.Slider(1, 12, value=4, step=1, label="Max. Number of Steps")
+                steps_input = gr.Slider(
+                    1, 12, value=4, step=1, label="Max. Number of Steps"
+                )
                 steps_input.change(self.agent_set_steps, steps_input, None)
-                tools_list = gr.Dropdown(self.agent_get_tools(), interactive=True, label="Tools", info="(display only)")
+                tools_list = gr.Dropdown(
+                    self.agent_get_tools(),
+                    interactive=True,
+                    label="Tools",
+                    info="(display only)",
+                )
                 # tools_list.select(self.agent_get_tools, None, tools_list)
                 reset = gr.Button(value="Reset ChronoCore-77")
                 reset.click(self.agent_reset, None, None)
-            text_input = gr.Textbox(lines=1, label="Chat Message", elem_classes="cyber-glitch-2", max_length=1000)
+            text_input = gr.Textbox(
+                lines=1,
+                label="Chat Message",
+                elem_classes="cyber-glitch-2",
+                max_length=1000,
+            )
             text_input.submit(
                 self.log_user_message,
                 [text_input, file_uploads_log],
                 [stored_messages, text_input],
             ).then(self.interact_with_agent, [stored_messages, chatbot], [chatbot])
             examples = gr.Examples(
-                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"], ["Compute the current local time glitch coeficients"],["Generate a bold picture inspired by the current local time"]],
-            inputs=[text_input],
+                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"],
+                    ["Compute the current local time glitch coeficients"],
+                    ["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"]
+        demo.launch(
+            debug=True,
+            share=True,
+            ssr_mode=False,
+            allowed_paths=["Cyberpunk.otf"],
+            **kwargs,
+        )
+
+
+__all__ = ["stream_to_gradio", "GradioUI"]