kpbotla commited on
Commit
53dc771
·
verified ·
1 Parent(s): 81917a3

Upload 2 files

Browse files
Files changed (2) hide show
  1. app.py +90 -124
  2. requirements.txt +6 -1
app.py CHANGED
@@ -1,103 +1,131 @@
1
  import os
2
  import gradio as gr
3
  import requests
4
- import inspect
5
  import pandas as pd
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6
 
7
- # (Keep Constants as is)
8
- # --- Constants ---
9
- DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 
 
10
 
11
  # --- Basic Agent Definition ---
12
- # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
13
  class BasicAgent:
14
  def __init__(self):
15
- print("BasicAgent initialized.")
 
 
 
 
 
 
 
 
 
 
16
  def __call__(self, question: str) -> str:
17
- print(f"Agent received question (first 50 chars): {question[:50]}...")
18
- fixed_answer = "This is a default answer."
19
- print(f"Agent returning fixed answer: {fixed_answer}")
20
- return fixed_answer
21
-
22
- def run_and_submit_all( profile: gr.OAuthProfile | None):
23
- """
24
- Fetches all questions, runs the BasicAgent on them, submits all answers,
25
- and displays the results.
26
- """
27
- # --- Determine HF Space Runtime URL and Repo URL ---
28
- space_id = os.getenv("SPACE_ID") # Get the SPACE_ID for sending link to the code
29
 
 
 
30
  if profile:
31
- username= f"{profile.username}"
32
- print(f"User logged in: {username}")
33
  else:
34
- print("User not logged in.")
35
  return "Please Login to Hugging Face with the button.", None
36
 
37
  api_url = DEFAULT_API_URL
38
  questions_url = f"{api_url}/questions"
39
  submit_url = f"{api_url}/submit"
40
 
41
- # 1. Instantiate Agent ( modify this part to create your agent)
42
  try:
43
  agent = BasicAgent()
44
  except Exception as e:
45
- print(f"Error instantiating agent: {e}")
46
  return f"Error initializing agent: {e}", None
47
- # 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)
48
  agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
49
- print(agent_code)
50
 
51
- # 2. Fetch Questions
52
- print(f"Fetching questions from: {questions_url}")
53
  try:
54
  response = requests.get(questions_url, timeout=15)
55
  response.raise_for_status()
56
  questions_data = response.json()
57
- if not questions_data:
58
- print("Fetched questions list is empty.")
59
- return "Fetched questions list is empty or invalid format.", None
60
- print(f"Fetched {len(questions_data)} questions.")
61
- except requests.exceptions.RequestException as e:
62
- print(f"Error fetching questions: {e}")
63
- return f"Error fetching questions: {e}", None
64
- except requests.exceptions.JSONDecodeError as e:
65
- print(f"Error decoding JSON response from questions endpoint: {e}")
66
- print(f"Response text: {response.text[:500]}")
67
- return f"Error decoding server response for questions: {e}", None
68
  except Exception as e:
69
- print(f"An unexpected error occurred fetching questions: {e}")
70
- return f"An unexpected error occurred fetching questions: {e}", None
71
 
72
- # 3. Run your Agent
73
  results_log = []
74
  answers_payload = []
75
- print(f"Running agent on {len(questions_data)} questions...")
76
  for item in questions_data:
77
  task_id = item.get("task_id")
78
  question_text = item.get("question")
79
  if not task_id or question_text is None:
80
- print(f"Skipping item with missing task_id or question: {item}")
81
  continue
82
  try:
83
  submitted_answer = agent(question_text)
84
  answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
85
  results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
86
  except Exception as e:
87
- print(f"Error running agent on task {task_id}: {e}")
88
- results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": f"AGENT ERROR: {e}"})
89
 
90
  if not answers_payload:
91
- print("Agent did not produce any answers to submit.")
92
  return "Agent did not produce any answers to submit.", pd.DataFrame(results_log)
93
 
94
- # 4. Prepare Submission
95
- submission_data = {"username": username.strip(), "agent_code": agent_code, "answers": answers_payload}
96
- status_update = f"Agent finished. Submitting {len(answers_payload)} answers for user '{username}'..."
97
- print(status_update)
 
98
 
99
- # 5. Submit
100
- print(f"Submitting {len(answers_payload)} answers to: {submit_url}")
101
  try:
102
  response = requests.post(submit_url, json=submission_data, timeout=60)
103
  response.raise_for_status()
@@ -109,88 +137,26 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
109
  f"({result_data.get('correct_count', '?')}/{result_data.get('total_attempted', '?')} correct)\n"
110
  f"Message: {result_data.get('message', 'No message received.')}"
111
  )
112
- print("Submission successful.")
113
  results_df = pd.DataFrame(results_log)
114
  return final_status, results_df
115
- except requests.exceptions.HTTPError as e:
116
- error_detail = f"Server responded with status {e.response.status_code}."
117
- try:
118
- error_json = e.response.json()
119
- error_detail += f" Detail: {error_json.get('detail', e.response.text)}"
120
- except requests.exceptions.JSONDecodeError:
121
- error_detail += f" Response: {e.response.text[:500]}"
122
- status_message = f"Submission Failed: {error_detail}"
123
- print(status_message)
124
- results_df = pd.DataFrame(results_log)
125
- return status_message, results_df
126
- except requests.exceptions.Timeout:
127
- status_message = "Submission Failed: The request timed out."
128
- print(status_message)
129
- results_df = pd.DataFrame(results_log)
130
- return status_message, results_df
131
- except requests.exceptions.RequestException as e:
132
- status_message = f"Submission Failed: Network error - {e}"
133
- print(status_message)
134
- results_df = pd.DataFrame(results_log)
135
- return status_message, results_df
136
  except Exception as e:
137
- status_message = f"An unexpected error occurred during submission: {e}"
138
- print(status_message)
139
- results_df = pd.DataFrame(results_log)
140
- return status_message, results_df
141
-
142
 
143
- # --- Build Gradio Interface using Blocks ---
144
  with gr.Blocks() as demo:
145
- gr.Markdown("# Basic Agent Evaluation Runner")
146
- gr.Markdown(
147
- """
148
- **Instructions:**
149
-
150
- 1. Please clone this space, then modify the code to define your agent's logic, the tools, the necessary packages, etc ...
151
- 2. Log in to your Hugging Face account using the button below. This uses your HF username for submission.
152
- 3. Click 'Run Evaluation & Submit All Answers' to fetch questions, run your agent, submit answers, and see the score.
153
-
154
- ---
155
- **Disclaimers:**
156
- Once clicking on the "submit button, it can take quite some time ( this is the time for the agent to go through all the questions).
157
- This space provides a basic setup and is intentionally sub-optimal to encourage you to develop your own, more robust solution. For instance for the delay process of the submit button, a solution could be to cache the answers and submit in a seperate action or even to answer the questions in async.
158
- """
159
- )
160
 
161
  gr.LoginButton()
162
-
163
  run_button = gr.Button("Run Evaluation & Submit All Answers")
164
-
165
  status_output = gr.Textbox(label="Run Status / Submission Result", lines=5, interactive=False)
166
- # Removed max_rows=10 from DataFrame constructor
167
  results_table = gr.DataFrame(label="Questions and Agent Answers", wrap=True)
168
 
169
- run_button.click(
170
- fn=run_and_submit_all,
171
- outputs=[status_output, results_table]
172
- )
173
 
174
  if __name__ == "__main__":
175
- print("\n" + "-"*30 + " App Starting " + "-"*30)
176
- # Check for SPACE_HOST and SPACE_ID at startup for information
177
- space_host_startup = os.getenv("SPACE_HOST")
178
- space_id_startup = os.getenv("SPACE_ID") # Get SPACE_ID at startup
179
-
180
- if space_host_startup:
181
- print(f"✅ SPACE_HOST found: {space_host_startup}")
182
- print(f" Runtime URL should be: https://{space_host_startup}.hf.space")
183
- else:
184
- print("ℹ️ SPACE_HOST environment variable not found (running locally?).")
185
-
186
- if space_id_startup: # Print repo URLs if SPACE_ID is found
187
- print(f"✅ SPACE_ID found: {space_id_startup}")
188
- print(f" Repo URL: https://huggingface.co/spaces/{space_id_startup}")
189
- print(f" Repo Tree URL: https://huggingface.co/spaces/{space_id_startup}/tree/main")
190
- else:
191
- print("ℹ️ SPACE_ID environment variable not found (running locally?). Repo URL cannot be determined.")
192
-
193
- print("-"*(60 + len(" App Starting ")) + "\n")
194
-
195
- print("Launching Gradio Interface for Basic Agent Evaluation...")
196
- demo.launch(debug=True, share=False)
 
1
  import os
2
  import gradio as gr
3
  import requests
 
4
  import pandas as pd
5
+ from smolagents import Agent, tool
6
+ from duckduckgo_search import DDGS
7
+ from transformers import pipeline
8
+
9
+ # --- Tool Definitions ---
10
+ @tool
11
+ class WebSearchTool:
12
+ name = "web_search"
13
+ description = "Search the web for up-to-date factual information."
14
+ def use(self, query: str) -> str:
15
+ with DDGS() as ddgs:
16
+ results = ddgs.text(query)
17
+ output = [f"{r['title']} - {r['href']}" for r in results[:3]]
18
+ return "\n".join(output) if output else "No relevant results found."
19
+
20
+ @tool
21
+ class CiteTool:
22
+ name = "cite"
23
+ description = "Add citation to a given answer with a valid URL."
24
+ def use(self, input: str) -> str:
25
+ try:
26
+ answer, url = input.split("|||")
27
+ return f"{answer.strip()}\n\nSource: [{url.strip()}]({url.strip()})"
28
+ except:
29
+ return "Could not format citation correctly."
30
+
31
+ summarizer = pipeline("summarization")
32
+ @tool
33
+ class SummarizerTool:
34
+ name = "summarize"
35
+ description = "Summarize a long text into a short paragraph."
36
+ def use(self, input: str) -> str:
37
+ if len(input) < 50:
38
+ return input
39
+ result = summarizer(input, max_length=100, min_length=25, do_sample=False)
40
+ return result[0]['summary_text']
41
+
42
+ @tool
43
+ class PythonTool:
44
+ name = "python"
45
+ description = "Execute Python code to solve math problems."
46
+ def use(self, code: str) -> str:
47
+ try:
48
+ result = str(eval(code, {"__builtins__": {}}))
49
+ return f"Answer: {result}"
50
+ except Exception as e:
51
+ return f"Error: {str(e)}"
52
 
53
+ @tool
54
+ class FallbackTool:
55
+ name = "fallback"
56
+ description = "Handle unanswerable or unclear queries."
57
+ def use(self, _: str) -> str:
58
+ return "I'm sorry, I couldn't find the answer to your question. Could you rephrase or try something else?"
59
 
60
  # --- Basic Agent Definition ---
 
61
  class BasicAgent:
62
  def __init__(self):
63
+ tools = [WebSearchTool(), CiteTool(), SummarizerTool(), PythonTool(), FallbackTool()]
64
+ self.agent = Agent(
65
+ tools=tools,
66
+ system_prompt="""
67
+ You are Smart Answering Agent v3.
68
+ Answer questions factually, concisely, and cite sources when available.
69
+ Route to the correct tool for factual, math, or summarization queries.
70
+ If you don’t know the answer, respond gracefully using the fallback tool.
71
+ Ensure output format is friendly for the GAIA evaluation.
72
+ """
73
+ )
74
  def __call__(self, question: str) -> str:
75
+ return self.agent.run(question)
76
+
77
+ # --- Evaluation Logic ---
78
+ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 
 
 
 
 
 
 
79
 
80
+ def run_and_submit_all(profile: gr.OAuthProfile | None):
81
+ space_id = os.getenv("SPACE_ID")
82
  if profile:
83
+ username = profile.username
 
84
  else:
 
85
  return "Please Login to Hugging Face with the button.", None
86
 
87
  api_url = DEFAULT_API_URL
88
  questions_url = f"{api_url}/questions"
89
  submit_url = f"{api_url}/submit"
90
 
 
91
  try:
92
  agent = BasicAgent()
93
  except Exception as e:
 
94
  return f"Error initializing agent: {e}", None
95
+
96
  agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
 
97
 
 
 
98
  try:
99
  response = requests.get(questions_url, timeout=15)
100
  response.raise_for_status()
101
  questions_data = response.json()
 
 
 
 
 
 
 
 
 
 
 
102
  except Exception as e:
103
+ return f"Error fetching questions: {e}", None
 
104
 
 
105
  results_log = []
106
  answers_payload = []
107
+
108
  for item in questions_data:
109
  task_id = item.get("task_id")
110
  question_text = item.get("question")
111
  if not task_id or question_text is None:
 
112
  continue
113
  try:
114
  submitted_answer = agent(question_text)
115
  answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
116
  results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
117
  except Exception as e:
118
+ results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": f"AGENT ERROR: {e}"})
 
119
 
120
  if not answers_payload:
 
121
  return "Agent did not produce any answers to submit.", pd.DataFrame(results_log)
122
 
123
+ submission_data = {
124
+ "username": username.strip(),
125
+ "agent_code": agent_code,
126
+ "answers": answers_payload
127
+ }
128
 
 
 
129
  try:
130
  response = requests.post(submit_url, json=submission_data, timeout=60)
131
  response.raise_for_status()
 
137
  f"({result_data.get('correct_count', '?')}/{result_data.get('total_attempted', '?')} correct)\n"
138
  f"Message: {result_data.get('message', 'No message received.')}"
139
  )
 
140
  results_df = pd.DataFrame(results_log)
141
  return final_status, results_df
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
142
  except Exception as e:
143
+ return f"Submission Failed: {e}", pd.DataFrame(results_log)
 
 
 
 
144
 
145
+ # --- Gradio UI ---
146
  with gr.Blocks() as demo:
147
+ gr.Markdown("# Smart Agent Evaluation Runner")
148
+ gr.Markdown("""
149
+ **Instructions:**
150
+ 1. Login to your HF account using the button.
151
+ 2. Click 'Run Evaluation & Submit All Answers' to test your agent.
152
+ """)
 
 
 
 
 
 
 
 
 
153
 
154
  gr.LoginButton()
 
155
  run_button = gr.Button("Run Evaluation & Submit All Answers")
 
156
  status_output = gr.Textbox(label="Run Status / Submission Result", lines=5, interactive=False)
 
157
  results_table = gr.DataFrame(label="Questions and Agent Answers", wrap=True)
158
 
159
+ run_button.click(fn=run_and_submit_all, outputs=[status_output, results_table])
 
 
 
160
 
161
  if __name__ == "__main__":
162
+ demo.launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
requirements.txt CHANGED
@@ -1,2 +1,7 @@
 
 
 
 
1
  gradio
2
- requests
 
 
1
+ smolagents
2
+ duckduckgo-search
3
+ transformers
4
+ torch
5
  gradio
6
+ requests
7
+ pandas