Add HASHIRU logo and update UI header in main.py

main.py

@@ -3,13 +3,30 @@ from src.manager import GeminiManager
 from src.tool_loader import ToolLoader
 import gradio as gr
 import time
+import base64
+
+_logo_bytes = open("HASHIRU_LOGO.png", "rb").read()
+_logo_b64 = base64.b64encode(_logo_bytes).decode()
+_header_html = f"""
+<div style="
+    display: flex;
+    flex-direction: column;
+    align-items: center;
+    padding-right: 24px;
+">
+  <img src="data:image/png;base64,{_logo_b64}" width="20" height="20" />
+  <span style="margin-top: 8px; font-size: 20px; font-weight: bold; color: white;">
+    HASHIRU AI
+  </span>
+</div>
+"""
 
 if __name__ == "__main__":
     # Define the tool metadata for orchestration.
     # Load the tools using the ToolLoader class.
     tool_loader = ToolLoader()
 
-    model_manager = GeminiManager(toolsLoader=tool_loader, gemini_model="gemini-2.5-pro-preview-03-25")
+    model_manager = GeminiManager(toolsLoader=tool_loader, gemini_model="gemini-2.0-flash")
     
     def user_message(msg: str, history: list) -> tuple[str, list]:
         """Adds user message to chat history"""
@@ -29,10 +46,33 @@ if __name__ == "__main__":
         new_history[-1]['content'] = edit_data.value
         # yield new_history, gr.update(interactive=False,)
         yield from model_manager.run(new_history)
+
+    def update_model(model_name):
+        print(f"Model changed to: {model_name}")
+        pass
     
-    with gr.Blocks(fill_width=True, fill_height=True) as demo:
-        gr.Markdown("# Hashiru AI")
-        
+    css = """
+    #title-row { background: #2c2c2c; border-radius: 8px; padding: 8px; }
+    """
+    with gr.Blocks(css=css) as demo:
+        with gr.Row():
+            gr.HTML(_header_html)
+            model_dropdown = gr.Dropdown(
+                    choices=[
+                        "HASHIRU",
+                        "Static-HASHIRU",
+                        "Cloud-Only HASHIRU",
+                        "Local-Only HASHIRU",
+                        "No-Economy HASHIRU",
+                    ],
+                    value="HASHIRU",
+                    # label="HASHIRU",
+                    scale=4,
+                    interactive=True,
+            )
+
+        model_dropdown.change(fn=update_model, inputs=model_dropdown, outputs=[])
+
         chatbot = gr.Chatbot(
             avatar_images=("HASHIRU_2.png", "HASHIRU.png"),
             type="messages",