Update app.py

app.py

@@ -3,25 +3,75 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
+from smolagents import CodeAgent, LiteLLMModel, DuckDuckGoSearchTool, VisitWebpageTool, FinalAnswerTool
 
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
-# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
+# ----- THIS IS WHERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
+        
+        # Create model using Hugging Face Inference API
+        # This will use the HF_TOKEN environment variable if available
+        try:
+            # Using a good open model available on HF Inference API
+            self.model = LiteLLMModel(
+                model_id="huggingface/meta-llama/Llama-3.2-3B-Instruct",
+                api_base="https://api-inference.huggingface.co/models/",
+                # HF_TOKEN will be automatically used if set in environment
+                max_tokens=1024
+            )
+            print("Initialized Hugging Face Inference API model")
+        except Exception as e:
+            print(f"Error initializing HF model: {e}")
+            # Fallback to a simpler approach if model fails
+            self.model = None
+        
+        # Create the CodeAgent with tools if model is available
+        if self.model:
+            try:
+                self.agent = CodeAgent(
+                    tools=[
+                        DuckDuckGoSearchTool(),
+                        VisitWebpageTool(),
+                        FinalAnswerTool()
+                    ],
+                    model=self.model,
+                    additional_authorized_imports=["wikipedia", "requests", "json", "re", "datetime", "os"]
+                )
+                print("CodeAgent with tools initialized successfully")
+            except Exception as e:
+                print(f"Error initializing CodeAgent: {e}")
+                self.agent = None
+        else:
+            self.agent = None
+    
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = "This is a default answer."
-        print(f"Agent returning fixed answer: {fixed_answer}")
-        return fixed_answer
-
-def run_and_submit_all( profile: gr.OAuthProfile | None):
+        
+        if self.agent:
+            try:
+                # Use smolagents CodeAgent to process the question
+                answer = self.agent.run(question)
+                print(f"Agent returning answer: {answer[:100]}...")
+                return answer
+            except Exception as e:
+                print(f"Error in agent processing: {e}")
+                # Return a fallback answer in case of errors
+                return f"Error processing question: {str(e)}"
+        else:
+            # Fallback if agent initialization failed
+            fallback_answer = "Agent initialization failed. Please check model configuration."
+            print(f"Agent returning fallback answer: {fallback_answer}")
+            return fallback_answer
+
+def run_and_submit_all(profile: gr.OAuthProfile | None):
     """
-    Fetches all questions, runs the BasicAgent on them, submits all answers,
+    Fetches all questions, runs the SmolagentsAgent on them, submits all answers,
     and displays the results.
     """
     # --- Determine HF Space Runtime URL and Repo URL ---
@@ -38,13 +88,14 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     questions_url = f"{api_url}/questions"
     submit_url = f"{api_url}/submit"
 
-    # 1. Instantiate Agent ( modify this part to create your agent)
+    # 1. Instantiate Agent (modify this part to create your agent)
     try:
         agent = BasicAgent()
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None
-    # In the case of an app running as a hugging Face space, this link points toward your codebase ( usefull for others so please keep it public)
+    
+    # In the case of an app running as a hugging Face space, this link points toward your codebase
     agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
     print(agent_code)