Update app.py

app.py

@@ -17,7 +17,10 @@ import soundfile as sf
 import numpy as np
 from requests.adapters import HTTPAdapter
 from urllib3.util.retry import Retry
-from huggingface_hub import InferenceClient, get_inference_client, Agent
+
+# CORRECTED IMPORTS:
+from huggingface_hub import InferenceClient
+from huggingface_hub.agents import Agent # Corrected import path for Agent
 
 # Setup log file path
 log_path = os.path.join(os.getcwd(), f"agent_log_{datetime.now().strftime('%Y%m%d_%H%M%S')}.log")
@@ -124,14 +127,8 @@ class SmartAgent:
     def _initialize_agent(self) -> Agent:
         logging.info("Initializing Hugging Face Agent...")
         try:
-            # Ensure HF_TOKEN is available in the environment or via login
-            hf_token = os.environ.get("HF_TOKEN")
-            if not hf_token:
-                # Fallback to get_inference_client's default token handling (e.g., from ~/.cache/huggingface/token)
-                logging.warning("HF_TOKEN environment variable not set. Relying on Hugging Face Hub client token.")
-                client = get_inference_client() # This will pick up token from env or cache
-            else:
-                client = InferenceClient(token=hf_token)
+            # InferenceClient will automatically pick up HF_TOKEN from env or cache
+            client = InferenceClient() 
 
             # Check if agent already exists for this username
             list_agents_resp = self.http_session.get(f"{API_URL}/agents")
@@ -271,6 +268,7 @@ class SmartAgent:
         try:
             if tool_name == "image-to-text":
                 image_bytes = self.http_session.get(inputs["image"]).content
+                # Use NamedTemporaryFile for safer unique file handling
                 with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmp_img_file:
                     tmp_img_file.write(image_bytes)
                     image_path = tmp_img_file.name
@@ -340,14 +338,9 @@ class SmartAgent:
         logging.info("Cleaning up agent resources...")
         if self.agent and self.agent_id:
             try:
-                # Assuming the Hugging Face Agent object has a delete method or similar
-                # For this course, you might need to call a specific API endpoint or rely on
-                # the agent session expiring if no direct delete method is provided by the lib.
-                # If there's a specific API endpoint for deleting agents, use it here.
-                # For now, we'll log its ID as deleted.
-                # If agent.delete() exists, use it:
-                # self.agent.delete()
-                # For the course, simply logging and clearing internal state might be enough.
+                # If the Hugging Face Agent library provides a direct delete method, use it.
+                # Otherwise, the agent will typically expire after inactivity.
+                # Example if agent.delete() exists: self.agent.delete()
                 logging.info(f"Agent with ID {self.agent_id} is conceptually deleted (or will expire).")
             except Exception as e:
                 logging.warning(f"Failed to delete agent or clean up its remote state: {e}")