Update app.py
Browse files
app.py
CHANGED
@@ -3,6 +3,8 @@ import gradio as gr
|
|
3 |
import requests
|
4 |
import inspect
|
5 |
import pandas as pd
|
|
|
|
|
6 |
|
7 |
# (Keep Constants as is)
|
8 |
# --- Constants ---
|
@@ -10,14 +12,113 @@ 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 |
-
|
15 |
print("BasicAgent initialized.")
|
|
|
16 |
def __call__(self, question: str) -> str:
|
17 |
print(f"Agent received question (first 50 chars): {question[:50]}...")
|
18 |
-
|
19 |
-
|
20 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
21 |
|
22 |
def run_and_submit_all( profile: gr.OAuthProfile | None):
|
23 |
"""
|
@@ -91,7 +192,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
|
|
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)
|
@@ -138,6 +239,15 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
|
|
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 ---
|
@@ -193,4 +303,5 @@ if __name__ == "__main__":
|
|
193 |
print("-"*(60 + len(" App Starting ")) + "\n")
|
194 |
|
195 |
print("Launching Gradio Interface for Basic Agent Evaluation...")
|
196 |
-
demo.launch(debug=True, share=False)
|
|
|
|
3 |
import requests
|
4 |
import inspect
|
5 |
import pandas as pd
|
6 |
+
from bs4 import BeautifulSoup
|
7 |
+
import requests
|
8 |
|
9 |
# (Keep Constants as is)
|
10 |
# --- Constants ---
|
|
|
12 |
|
13 |
# --- Basic Agent Definition ---
|
14 |
# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
|
15 |
+
|
16 |
+
|
17 |
+
def web_search(query: str) -> list[dict]:
|
18 |
+
"""
|
19 |
+
Performs a web search and returns relevant information.
|
20 |
+
|
21 |
+
Args:
|
22 |
+
query: The search query string.
|
23 |
+
|
24 |
+
Returns:
|
25 |
+
A list of dictionaries, where each dictionary represents a search result
|
26 |
+
with keys 'title', 'snippet', and 'url'. Returns an empty list if no
|
27 |
+
results are found or an error occurs.
|
28 |
+
"""
|
29 |
+
search_url = f"https://www.google.com/search?q={requests.utils.quote(query)}"
|
30 |
+
headers = {
|
31 |
+
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"
|
32 |
+
}
|
33 |
+
results = []
|
34 |
+
|
35 |
+
try:
|
36 |
+
response = requests.get(search_url, headers=headers, timeout=10)
|
37 |
+
response.raise_for_status() # Raise an HTTPError for bad responses (4xx or 5xx)
|
38 |
+
|
39 |
+
soup = BeautifulSoup(response.text, 'html.parser')
|
40 |
+
|
41 |
+
# Find search results - this is a basic example and might need adjustment
|
42 |
+
# based on Google's ever-changing HTML structure.
|
43 |
+
# 'div.tF2CMy' is a common class for result blocks as of certain dates.
|
44 |
+
search_results = soup.select('div.tF2CMy')
|
45 |
+
|
46 |
+
if not search_results:
|
47 |
+
# Fallback or alternative selectors if the primary one fails
|
48 |
+
search_results = soup.select('div.g') # Another common class
|
49 |
+
|
50 |
+
for result in search_results:
|
51 |
+
link = result.select_one('a')
|
52 |
+
title = result.select_one('h3')
|
53 |
+
snippet = result.select_one('span.aCOpNe') # Example snippet class
|
54 |
+
|
55 |
+
if link and title:
|
56 |
+
item = {
|
57 |
+
'title': title.get_text(),
|
58 |
+
'url': link['href'],
|
59 |
+
'snippet': snippet.get_text() if snippet else 'No snippet available'
|
60 |
+
}
|
61 |
+
results.append(item)
|
62 |
+
|
63 |
+
except requests.exceptions.RequestException as e:
|
64 |
+
print(f"Error during web search request: {e}")
|
65 |
+
except Exception as e:
|
66 |
+
print(f"An unexpected error occurred during web search: {e}")
|
67 |
+
|
68 |
+
return results
|
69 |
+
|
70 |
+
# Example usage (optional, for testing)
|
71 |
+
# search_results = web_search("what is the capital of France?")
|
72 |
+
# for i, result in enumerate(search_results[:3]): # Print first 3 results
|
73 |
+
# print(f"--- Result {i+1} ---")
|
74 |
+
# print(f"Title: {result.get('title', 'N/A')}")
|
75 |
+
# print(f"URL: {result.get('url', 'N/A')}")
|
76 |
+
# print(f"Snippet: {result.get('snippet', 'N/A')}")
|
77 |
+
# print("-" * 10)
|
78 |
+
|
79 |
class BasicAgent:
|
80 |
+
def __init__(self):
|
81 |
print("BasicAgent initialized.")
|
82 |
+
|
83 |
def __call__(self, question: str) -> str:
|
84 |
print(f"Agent received question (first 50 chars): {question[:50]}...")
|
85 |
+
|
86 |
+
# Simple logic to determine if a web search is needed
|
87 |
+
question_lower = question.lower()
|
88 |
+
search_keywords = ["what is", "how to", "where is", "who is", "when did", "define", "explain", "tell me about"]
|
89 |
+
needs_search = any(keyword in question_lower for keyword in search_keywords) or "?" in question
|
90 |
+
|
91 |
+
if needs_search:
|
92 |
+
print(f"Question likely requires search. Searching for: {question}")
|
93 |
+
search_results = web_search(question) # Call the web_search function
|
94 |
+
|
95 |
+
if search_results:
|
96 |
+
# Process search results to formulate an answer
|
97 |
+
answer_parts = []
|
98 |
+
for i, result in enumerate(search_results[:3]): # Use top 3 results
|
99 |
+
if result.get('snippet'):
|
100 |
+
answer_parts.append(f"Snippet {i+1}: {result['snippet']}")
|
101 |
+
elif result.get('title'):
|
102 |
+
answer_parts.append(f"Result {i+1} Title: {result['title']}")
|
103 |
+
|
104 |
+
|
105 |
+
if answer_parts:
|
106 |
+
formulated_answer = "Based on web search:\n" + "\n".join(answer_parts)
|
107 |
+
print(f"Agent returning search-based answer: {formulated_answer[:100]}...")
|
108 |
+
return formulated_answer
|
109 |
+
else:
|
110 |
+
print("Web search returned results but no useful snippets/titles found.")
|
111 |
+
return "I couldn't find a specific answer from the web search results."
|
112 |
+
|
113 |
+
else:
|
114 |
+
print("Web search returned no results.")
|
115 |
+
return "I couldn't find any relevant information on the web for your question."
|
116 |
+
else:
|
117 |
+
# If no search is needed, return a default or simple response
|
118 |
+
print("Question does not appear to require search. Returning fixed answer.")
|
119 |
+
fixed_answer = "How can I help you?"
|
120 |
+
return fixed_answer
|
121 |
+
|
122 |
|
123 |
def run_and_submit_all( profile: gr.OAuthProfile | None):
|
124 |
"""
|
|
|
192 |
print("Agent did not produce any answers to submit.")
|
193 |
return "Agent did not produce any answers to submit.", pd.DataFrame(results_log)
|
194 |
|
195 |
+
# 4. Prepare Submission
|
196 |
submission_data = {"username": username.strip(), "agent_code": agent_code, "answers": answers_payload}
|
197 |
status_update = f"Agent finished. Submitting {len(answers_payload)} answers for user '{username}'..."
|
198 |
print(status_update)
|
|
|
239 |
print(status_message)
|
240 |
results_df = pd.DataFrame(results_log)
|
241 |
return status_message, results_df
|
242 |
+
|
243 |
+
def __init__(self):
|
244 |
+
print("BasicAgent initialized.")
|
245 |
+
def __call__(self, question: str) -> str:
|
246 |
+
print(f"Agent received question (first 50 chars): {question[:50]}...")
|
247 |
+
fixed_answer = "How i can help you?"
|
248 |
+
print(f"Agent returning fixed answer: {fixed_answer}")
|
249 |
+
return fixed_answer
|
250 |
+
|
251 |
|
252 |
|
253 |
# --- Build Gradio Interface using Blocks ---
|
|
|
303 |
print("-"*(60 + len(" App Starting ")) + "\n")
|
304 |
|
305 |
print("Launching Gradio Interface for Basic Agent Evaluation...")
|
306 |
+
demo.launch(debug=True, share=False)7
|
307 |
+
|