refactoring

.idea/.gitignore

@@ -0,0 +1,5 @@
+# Default ignored files
+/shelf/
+/workspace.xml
+# Editor-based HTTP Client requests
+/httpRequests/

app.py

@@ -1,18 +1,32 @@
-from smolagents import CodeAgent, HfApiModel, load_tool, tool
-import datetime
-import requests
-import pytz
-import yaml
+from __future__ import annotations
+
+import os
 import shutil
 from pathlib import Path
+
 import gradio as gr
-print("Gradio version:", gr.__version__)
+import yaml
+from smolagents import CodeAgent, HfApiModel, load_tool
+
 from tools.final_answer import FinalAnswerTool
 from tools.web_search import DuckDuckGoSearchTool
 from tools.visit_webpage import VisitWebpageTool
+from tools.disk_free import disk_free
+from tools.timezone_time import get_current_time_in_timezone
 from src.first_agent.ui import GradioUI
 
-import os
+
+print("Gradio version:", gr.__version__)
+
+# --- Load .env locally (HF Spaces will use Settings → Secrets instead) ---
+try:
+    from dotenv import load_dotenv
+
+    load_dotenv()
+except Exception:
+    pass
+
+# --- Required env ---
 if not os.getenv("HF_TOKEN"):
     raise RuntimeError(
         "HF_TOKEN is not set. "
@@ -26,81 +40,30 @@ if OUTPUT_DIR.exists():
     shutil.rmtree(OUTPUT_DIR)
 OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
 
-@tool
-def disk_free(path: str = "/") -> str:
-    """
-    Show total/used/free disk space for a given filesystem path.
-    Args:
-        path: Filesystem path to inspect (e.g., '/', '/home/user').
-    Returns:
-        A human-friendly string with total/used/free sizes.
-    """
-    try:
-        usage = shutil.disk_usage(path)  # returns (total, used, free) in bytes
-    except Exception as e:
-        return f"Error reading disk usage for path='{path}': {e}"
-
-    def fmt_bytes(n: int) -> str:
-        # binary units (KiB, MiB, GiB...)
-        units = ["B", "KiB", "MiB", "GiB", "TiB", "PiB"]
-        x = float(n)
-        i = 0
-        while x >= 1024.0 and i < len(units) - 1:
-            x /= 1024.0
-            i += 1
-        return f"{x:.2f} {units[i]}"
-
-    total = fmt_bytes(usage.total)
-    used = fmt_bytes(usage.used)
-    free = fmt_bytes(usage.free)
-    return f"Disk usage for '{path}': total={total}, used={used}, free={free}"
-
-@tool
-def get_current_time_in_timezone(timezone: str) -> str:
-    """A tool that fetches the current local time in a specified timezone.
-    Args:
-        timezone: A string representing a valid timezone (e.g., 'America/New_York').
-    """
-    try:
-        # Create timezone object
-        tz = pytz.timezone(timezone)
-        # Get current time in that timezone
-        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
-        return f"The current local time in {timezone} is: {local_time}"
-    except Exception as e:
-        return f"Error fetching time for timezone '{timezone}': {str(e)}"
-
+# Make FinalAnswerTool write into the same folder
+os.environ["FINAL_ANSWER_DIR"] = str(OUTPUT_DIR)
 
+# --- Tools ---
 final_answer = FinalAnswerTool()
 web_search = DuckDuckGoSearchTool(max_results=10)
 visit_webpage = VisitWebpageTool()
 
-import os
-try:
-    from dotenv import load_dotenv
-    load_dotenv()
-except Exception:
-    # On HF this is not necessary; there you can access env from Settings → Secrets
-    pass
-
-
-# If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
-# model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 
-
+# --- Model ---
 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",  # may be overloaded sometimes
+    custom_role_conversions=None,
 )
 
-
-# Import tool from Hub
+# Tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
-with open("prompts.yaml", 'r') as stream:
+# Prompts
+with open("prompts.yaml", "r", encoding="utf-8") as stream:
     prompt_templates = yaml.safe_load(stream)
-    
+
+# --- Agent ---
 agent = CodeAgent(
     model=model,
     tools=[
@@ -117,8 +80,8 @@ agent = CodeAgent(
     planning_interval=None,
     name=None,
     description=None,
-    prompt_templates=prompt_templates
+    prompt_templates=prompt_templates,
 )
 
-
-GradioUI(agent).launch()
+# On HF Spaces, share=True is not supported; force share=False
+GradioUI(agent).launch()

src/first_agent/__init__.py

@@ -0,0 +1,26 @@
+# src/first_agent/__init__.py
+
+"""
+first_agent package.
+
+Public API surface:
+- GradioUI: UI wrapper to launch agent in Gradio
+"""
+
+from __future__ import annotations
+
+from importlib.metadata import PackageNotFoundError, version
+
+try:
+    __version__ = version("first_agent")
+except PackageNotFoundError:
+    # When running from source without installation
+    __version__ = "0.0.0"
+
+# Keep imports lightweight: only re-export core public objects.
+from .ui import GradioUI  # noqa: E402
+
+__all__ = [
+    "GradioUI",
+    "__version__",
+]

src/first_agent/ui.py

@@ -295,7 +295,7 @@ class GradioUI:
                 [stored_messages, text_input],
             ).then(self.interact_with_agent, [stored_messages, chatbot], [chatbot])
 
-        demo.launch(debug=True, share=True, **kwargs)
+        demo.launch(debug=True, **kwargs)
 
 
 __all__ = ["stream_to_gradio", "GradioUI"]

tools/__init__.py

@@ -0,0 +1,7 @@
+from .disk_free import disk_free
+from .timezone_time import get_current_time_in_timezone
+
+__all__ = [
+    "disk_free",
+    "get_current_time_in_timezone",
+]

tools/disk_free.py

@@ -0,0 +1,35 @@
+from __future__ import annotations
+
+import shutil
+from smolagents import tool
+
+
+@tool
+def disk_free(path: str = "/") -> str:
+    """
+    Show total/used/free disk space for a given filesystem path.
+
+    Args:
+        path: Filesystem path to inspect (e.g., '/', '/home/user').
+
+    Returns:
+        A human-friendly string with total/used/free sizes.
+    """
+    try:
+        usage = shutil.disk_usage(path)  # (total, used, free) in bytes
+    except Exception as e:
+        return f"Error reading disk usage for path='{path}': {e}"
+
+    def fmt_bytes(n: int) -> str:
+        units = ["B", "KiB", "MiB", "GiB", "TiB", "PiB"]
+        x = float(n)
+        i = 0
+        while x >= 1024.0 and i < (len(units) - 1):
+            x /= 1024.0
+            i += 1
+        return f"{x:.2f} {units[i]}"
+
+    total = fmt_bytes(usage.total)
+    used = fmt_bytes(usage.used)
+    free = fmt_bytes(usage.free)
+    return f"Disk usage for '{path}': total={total}, used={used}, free={free}"

tools/timezone_time.py

@@ -0,0 +1,24 @@
+from __future__ import annotations
+
+import datetime
+import pytz
+from smolagents import tool
+
+
+@tool
+def get_current_time_in_timezone(timezone: str) -> str:
+    """
+    Fetch current local time in a specified timezone.
+
+    Args:
+        timezone: A valid timezone string (e.g., 'America/New_York').
+
+    Returns:
+        A human-readable string with local time.
+    """
+    try:
+        tz = pytz.timezone(timezone)
+        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
+        return f"The current local time in {timezone} is: {local_time}"
+    except Exception as e:
+        return f"Error fetching time for timezone '{timezone}': {e}"