Spaces:
Sleeping
Sleeping
Added Code
Browse files- agent.py +232 -0
- app.py +196 -195
- requirements.txt +13 -2
agent.py
ADDED
|
@@ -0,0 +1,232 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
########## Imports ############
|
| 2 |
+
from langchain_google_genai import ChatGoogleGenerativeAI
|
| 3 |
+
import os
|
| 4 |
+
from typing import TypedDict, List, Dict, Any, Optional
|
| 5 |
+
from langgraph.graph import StateGraph, START, END
|
| 6 |
+
from langchain_openai import ChatOpenAI
|
| 7 |
+
from langchain_core.messages import HumanMessage
|
| 8 |
+
|
| 9 |
+
from langchain_community.tools import WikipediaQueryRun
|
| 10 |
+
from langchain_community.utilities import WikipediaAPIWrapper
|
| 11 |
+
import string
|
| 12 |
+
|
| 13 |
+
from langchain_experimental.tools import PythonREPLTool
|
| 14 |
+
import ast, json
|
| 15 |
+
|
| 16 |
+
from langchain_community.tools import DuckDuckGoSearchRun
|
| 17 |
+
|
| 18 |
+
|
| 19 |
+
########## State ############
|
| 20 |
+
class InfoState(TypedDict):
|
| 21 |
+
question: str
|
| 22 |
+
answer_type: Optional[str] # WebInfo - WIKI - MATH
|
| 23 |
+
answer_code : Optional[str]
|
| 24 |
+
main_parts: Optional[List[str]]
|
| 25 |
+
tool_answer : Optional[list[str]]
|
| 26 |
+
final_answer : Optional[str]
|
| 27 |
+
|
| 28 |
+
|
| 29 |
+
|
| 30 |
+
######### Nodes ############
|
| 31 |
+
def get_wiki_relate(state: InfoState) -> InfoState:
|
| 32 |
+
"""
|
| 33 |
+
Tool to Get the wikipedia info from keywords extracted from preprocessing at main_parts.
|
| 34 |
+
|
| 35 |
+
Uses: Wikipedia API
|
| 36 |
+
Returns: tool_answer (summary)
|
| 37 |
+
"""
|
| 38 |
+
print("Using Wikipedia...")
|
| 39 |
+
# Create the Wikipedia utility
|
| 40 |
+
wiki = WikipediaAPIWrapper(
|
| 41 |
+
lang="en", # Wikipedia language
|
| 42 |
+
top_k_results=1, # how many results to fetch
|
| 43 |
+
doc_content_chars_max=2000
|
| 44 |
+
)
|
| 45 |
+
|
| 46 |
+
# Make a tool from it
|
| 47 |
+
wiki_tool = WikipediaQueryRun(api_wrapper=wiki)
|
| 48 |
+
|
| 49 |
+
wiki_answer = wiki_tool.run(" ".join(state["main_parts"]))
|
| 50 |
+
state['tool_answer'] = wiki_answer
|
| 51 |
+
return state
|
| 52 |
+
|
| 53 |
+
|
| 54 |
+
def execute_code(state: InfoState) -> InfoState :
|
| 55 |
+
"""Tool to calculate any math using python code or get current date time."""
|
| 56 |
+
print("Execut Code...")
|
| 57 |
+
python_tool = PythonREPLTool()
|
| 58 |
+
code = state["answer_code"]
|
| 59 |
+
state["tool_answer"]=python_tool.run(code)
|
| 60 |
+
return state
|
| 61 |
+
|
| 62 |
+
def get_code(state:InfoState) -> InfoState:
|
| 63 |
+
"""From prompt get the code to run."""
|
| 64 |
+
print("Getting Code (Gemini)...")
|
| 65 |
+
prompt = (
|
| 66 |
+
f"You are a strict code generator. "
|
| 67 |
+
f"Given the question: '{state['question']}', "
|
| 68 |
+
f"return ONLY valid Python code that computes the answer IF the question is about math, date, or time. "
|
| 69 |
+
f"Otherwise, return exactly: print('not valid')\n\n"
|
| 70 |
+
f"Rules:\n"
|
| 71 |
+
f"- Output ONLY the code or print('not valid')\n"
|
| 72 |
+
f"- No explanations, no markdown, no extra text\n"
|
| 73 |
+
f"- No quotes around the code\n"
|
| 74 |
+
f"- Use print() to show the result\n"
|
| 75 |
+
f"- Import modules only if needed (e.g. datetime, math)"
|
| 76 |
+
)
|
| 77 |
+
|
| 78 |
+
# 2️⃣ Call Gemini
|
| 79 |
+
model = ChatGoogleGenerativeAI(model="gemini-2.5-flash")
|
| 80 |
+
response = model.invoke([HumanMessage(content=prompt)]).content.strip()
|
| 81 |
+
|
| 82 |
+
state["answer_code"] = response
|
| 83 |
+
|
| 84 |
+
|
| 85 |
+
return state
|
| 86 |
+
|
| 87 |
+
|
| 88 |
+
|
| 89 |
+
def preprocess_text(state: dict) -> InfoState:
|
| 90 |
+
|
| 91 |
+
"""
|
| 92 |
+
Preprocess text to get the keywords to help get results directly from wikipedia.
|
| 93 |
+
|
| 94 |
+
Input: raw question
|
| 95 |
+
Output: main_parts (list of keywords)
|
| 96 |
+
"""
|
| 97 |
+
print("Preprocess text (Gemini)...")
|
| 98 |
+
# 1️⃣ Prepare the prompt
|
| 99 |
+
prompt = (
|
| 100 |
+
"Extract the most important content words (nouns, proper names, key concepts) from this question that would help find the best-matching Wikipedia article. "
|
| 101 |
+
"If the question is not in English; translate key terms to English for Wikipedia's English edition. "
|
| 102 |
+
"Ignore stopwords (like 'who', 'is', 'the', 'of', 'in', 'current', 'what'), filler words, and typos. "
|
| 103 |
+
"Focus on entities and topics that exist as Wikipedia page titles. "
|
| 104 |
+
"Correct obvious spelling mistakes and expand common abbreviations if needed for better Wikipedia matching.\n\n"
|
| 105 |
+
"Question: '" + state["question"] + "'\n\n"
|
| 106 |
+
"Output ONLY a valid JSON list of 1–4 corrected, title-cased strings (e.g. [\"President of the United States\", \"Joe Biden\"]). "
|
| 107 |
+
"No explanations, no markdown, no extra text, no quotes around words, no trailing commas."
|
| 108 |
+
)
|
| 109 |
+
|
| 110 |
+
# 2️⃣ Call Gemini
|
| 111 |
+
model = ChatGoogleGenerativeAI(model="gemini-2.5-flash")
|
| 112 |
+
response = model.invoke([HumanMessage(content=prompt)]).content.strip()
|
| 113 |
+
|
| 114 |
+
# 3️⃣ Try to safely parse
|
| 115 |
+
try:
|
| 116 |
+
# First, try JSON
|
| 117 |
+
state["main_parts"] = json.loads(response)
|
| 118 |
+
except json.JSONDecodeError:
|
| 119 |
+
try:
|
| 120 |
+
# If not JSON, try Python literal
|
| 121 |
+
state["main_parts"] = ast.literal_eval(response)
|
| 122 |
+
except Exception:
|
| 123 |
+
# If both fail, store fallback info
|
| 124 |
+
print("⚠️ Model returned invalid content:", response)
|
| 125 |
+
state["main_parts"] = []
|
| 126 |
+
|
| 127 |
+
return state
|
| 128 |
+
|
| 129 |
+
|
| 130 |
+
|
| 131 |
+
|
| 132 |
+
def get_answer(state: InfoState) -> InfoState :
|
| 133 |
+
"""
|
| 134 |
+
Final Node that returns the final answer organized.
|
| 135 |
+
|
| 136 |
+
Combines: tool_answer → final_answer
|
| 137 |
+
"""
|
| 138 |
+
print("Getting Answer (Gemini)...")
|
| 139 |
+
|
| 140 |
+
prompt = (
|
| 141 |
+
"Answer the question based on the context below. "
|
| 142 |
+
#"If the question cannot be answered using the information provided, answer with 'I don't know'. "
|
| 143 |
+
"Question: " + state["question"] +
|
| 144 |
+
"\nContext: " + str(state.get("tool_answer")) +
|
| 145 |
+
"\nRewrite answer in clearer, simple way."
|
| 146 |
+
)
|
| 147 |
+
model = ChatGoogleGenerativeAI(model="gemini-2.5-flash")
|
| 148 |
+
state["final_answer"] = (model.invoke([HumanMessage(content=prompt)]).content)
|
| 149 |
+
|
| 150 |
+
return state
|
| 151 |
+
|
| 152 |
+
def get_type(state: InfoState) -> InfoState:
|
| 153 |
+
"""Choose which tool to use based on question type (WIKI, SEARCH, CODE)."""
|
| 154 |
+
print("Getting Type (Gemini)...")
|
| 155 |
+
|
| 156 |
+
prompt = "According to the Question " +state["question"] + " Select the best tool to answer WIKI if it's informatative or science question, WebInfo if it need up to data news,MATH if math or date or time You're very serious,just give one word from given"
|
| 157 |
+
model = ChatGoogleGenerativeAI(model="gemini-2.5-flash")
|
| 158 |
+
state["answer_type"] = (model.invoke([HumanMessage(content=prompt)]).content)
|
| 159 |
+
|
| 160 |
+
return state
|
| 161 |
+
|
| 162 |
+
|
| 163 |
+
|
| 164 |
+
|
| 165 |
+
def get_search_results(state: InfoState) -> InfoState:
|
| 166 |
+
"""Tool to search web for results using DuckDuckGo."""
|
| 167 |
+
print("Searching...")
|
| 168 |
+
|
| 169 |
+
search = DuckDuckGoSearchRun()
|
| 170 |
+
|
| 171 |
+
# Simple text result
|
| 172 |
+
state['tool_answer'] = search.run(state["question"]) #" " .join(state["main_parts"]))
|
| 173 |
+
|
| 174 |
+
return state
|
| 175 |
+
|
| 176 |
+
|
| 177 |
+
def route(state: InfoState):
|
| 178 |
+
print(state["answer_type"])
|
| 179 |
+
return state["answer_type"]
|
| 180 |
+
|
| 181 |
+
|
| 182 |
+
|
| 183 |
+
################# Graph ################
|
| 184 |
+
def get_graph():
|
| 185 |
+
graph = StateGraph(InfoState)
|
| 186 |
+
|
| 187 |
+
# Add nodes
|
| 188 |
+
graph.add_node("get_wiki_relate", get_wiki_relate)
|
| 189 |
+
graph.add_node("preprocess_text", preprocess_text)
|
| 190 |
+
graph.add_node("get_answer", get_answer)
|
| 191 |
+
graph.add_node("get_type", get_type)
|
| 192 |
+
graph.add_node("get_search_results", get_search_results)
|
| 193 |
+
graph.add_node("execute_code", execute_code)
|
| 194 |
+
graph.add_node("get_code", get_code)
|
| 195 |
+
|
| 196 |
+
# Add edges
|
| 197 |
+
graph.add_edge(START, "preprocess_text")
|
| 198 |
+
graph.add_edge("preprocess_text", "get_type")
|
| 199 |
+
|
| 200 |
+
|
| 201 |
+
# Add conditional edges
|
| 202 |
+
graph.add_conditional_edges(
|
| 203 |
+
"get_type",
|
| 204 |
+
route,
|
| 205 |
+
{
|
| 206 |
+
"WebInfo": "get_search_results",
|
| 207 |
+
"WIKI": "get_wiki_relate",
|
| 208 |
+
"MATH": "get_code"
|
| 209 |
+
}
|
| 210 |
+
)
|
| 211 |
+
|
| 212 |
+
# Add final edges
|
| 213 |
+
graph.add_edge("get_search_results", "get_answer")
|
| 214 |
+
graph.add_edge("get_wiki_relate", "get_answer")
|
| 215 |
+
|
| 216 |
+
graph.add_edge("get_code", "execute_code")
|
| 217 |
+
graph.add_edge("execute_code", "get_answer")
|
| 218 |
+
|
| 219 |
+
graph.add_edge("get_answer", END)
|
| 220 |
+
|
| 221 |
+
|
| 222 |
+
# Compile the graph
|
| 223 |
+
compiled_graph = graph.compile()
|
| 224 |
+
return compiled_graph
|
| 225 |
+
|
| 226 |
+
def ask(compiled_graph,question):
|
| 227 |
+
legitimate_result = compiled_graph.invoke({
|
| 228 |
+
"question": question,
|
| 229 |
+
|
| 230 |
+
})
|
| 231 |
+
|
| 232 |
+
return legitimate_result['final_answer']
|
app.py
CHANGED
|
@@ -1,196 +1,197 @@
|
|
| 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 |
-
|
| 19 |
-
|
| 20 |
-
|
| 21 |
-
|
| 22 |
-
|
| 23 |
-
|
| 24 |
-
|
| 25 |
-
|
| 26 |
-
|
| 27 |
-
|
| 28 |
-
|
| 29 |
-
|
| 30 |
-
|
| 31 |
-
|
| 32 |
-
|
| 33 |
-
|
| 34 |
-
|
| 35 |
-
|
| 36 |
-
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
|
| 40 |
-
|
| 41 |
-
|
| 42 |
-
|
| 43 |
-
|
| 44 |
-
|
| 45 |
-
|
| 46 |
-
|
| 47 |
-
|
| 48 |
-
|
| 49 |
-
|
| 50 |
-
|
| 51 |
-
|
| 52 |
-
|
| 53 |
-
|
| 54 |
-
|
| 55 |
-
response.
|
| 56 |
-
|
| 57 |
-
|
| 58 |
-
|
| 59 |
-
|
| 60 |
-
|
| 61 |
-
|
| 62 |
-
|
| 63 |
-
|
| 64 |
-
|
| 65 |
-
|
| 66 |
-
print(f"
|
| 67 |
-
|
| 68 |
-
|
| 69 |
-
|
| 70 |
-
|
| 71 |
-
|
| 72 |
-
|
| 73 |
-
|
| 74 |
-
|
| 75 |
-
|
| 76 |
-
|
| 77 |
-
|
| 78 |
-
|
| 79 |
-
|
| 80 |
-
|
| 81 |
-
|
| 82 |
-
|
| 83 |
-
|
| 84 |
-
|
| 85 |
-
|
| 86 |
-
|
| 87 |
-
|
| 88 |
-
|
| 89 |
-
|
| 90 |
-
|
| 91 |
-
|
| 92 |
-
|
| 93 |
-
|
| 94 |
-
|
| 95 |
-
|
| 96 |
-
|
| 97 |
-
|
| 98 |
-
|
| 99 |
-
|
| 100 |
-
|
| 101 |
-
|
| 102 |
-
|
| 103 |
-
response.
|
| 104 |
-
|
| 105 |
-
|
| 106 |
-
|
| 107 |
-
f"
|
| 108 |
-
f"
|
| 109 |
-
f"
|
| 110 |
-
f"
|
| 111 |
-
|
| 112 |
-
|
| 113 |
-
|
| 114 |
-
|
| 115 |
-
|
| 116 |
-
|
| 117 |
-
|
| 118 |
-
|
| 119 |
-
|
| 120 |
-
|
| 121 |
-
|
| 122 |
-
|
| 123 |
-
|
| 124 |
-
|
| 125 |
-
|
| 126 |
-
|
| 127 |
-
|
| 128 |
-
|
| 129 |
-
|
| 130 |
-
|
| 131 |
-
|
| 132 |
-
|
| 133 |
-
|
| 134 |
-
|
| 135 |
-
|
| 136 |
-
|
| 137 |
-
|
| 138 |
-
|
| 139 |
-
|
| 140 |
-
|
| 141 |
-
|
| 142 |
-
|
| 143 |
-
|
| 144 |
-
|
| 145 |
-
|
| 146 |
-
gr.Markdown(
|
| 147 |
-
|
| 148 |
-
|
| 149 |
-
|
| 150 |
-
|
| 151 |
-
|
| 152 |
-
|
| 153 |
-
|
| 154 |
-
|
| 155 |
-
|
| 156 |
-
|
| 157 |
-
|
| 158 |
-
|
| 159 |
-
|
| 160 |
-
|
| 161 |
-
|
| 162 |
-
|
| 163 |
-
|
| 164 |
-
|
| 165 |
-
|
| 166 |
-
|
| 167 |
-
|
| 168 |
-
|
| 169 |
-
|
| 170 |
-
|
| 171 |
-
|
| 172 |
-
|
| 173 |
-
|
| 174 |
-
|
| 175 |
-
|
| 176 |
-
|
| 177 |
-
|
| 178 |
-
|
| 179 |
-
|
| 180 |
-
|
| 181 |
-
|
| 182 |
-
print(f"
|
| 183 |
-
|
| 184 |
-
|
| 185 |
-
|
| 186 |
-
|
| 187 |
-
|
| 188 |
-
print(f"
|
| 189 |
-
print(f" Repo
|
| 190 |
-
|
| 191 |
-
|
| 192 |
-
|
| 193 |
-
|
| 194 |
-
|
| 195 |
-
|
|
|
|
| 196 |
demo.launch(debug=True, share=False)
|
|
|
|
| 1 |
+
import os
|
| 2 |
+
import gradio as gr
|
| 3 |
+
import requests
|
| 4 |
+
import inspect
|
| 5 |
+
import pandas as pd
|
| 6 |
+
from agent import get_graph,ask
|
| 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 |
+
compiled_graph = get_graph()
|
| 19 |
+
fixed_answer = ask(compiled_graph,question)
|
| 20 |
+
print(f"Agent returning fixed answer: {fixed_answer}")
|
| 21 |
+
return fixed_answer
|
| 22 |
+
|
| 23 |
+
def run_and_submit_all( profile: gr.OAuthProfile | None):
|
| 24 |
+
"""
|
| 25 |
+
Fetches all questions, runs the BasicAgent on them, submits all answers,
|
| 26 |
+
and displays the results.
|
| 27 |
+
"""
|
| 28 |
+
# --- Determine HF Space Runtime URL and Repo URL ---
|
| 29 |
+
space_id = os.getenv("SPACE_ID") # Get the SPACE_ID for sending link to the code
|
| 30 |
+
|
| 31 |
+
if profile:
|
| 32 |
+
username= f"{profile.username}"
|
| 33 |
+
print(f"User logged in: {username}")
|
| 34 |
+
else:
|
| 35 |
+
print("User not logged in.")
|
| 36 |
+
return "Please Login to Hugging Face with the button.", None
|
| 37 |
+
|
| 38 |
+
api_url = DEFAULT_API_URL
|
| 39 |
+
questions_url = f"{api_url}/questions"
|
| 40 |
+
submit_url = f"{api_url}/submit"
|
| 41 |
+
|
| 42 |
+
# 1. Instantiate Agent ( modify this part to create your agent)
|
| 43 |
+
try:
|
| 44 |
+
agent = BasicAgent()
|
| 45 |
+
except Exception as e:
|
| 46 |
+
print(f"Error instantiating agent: {e}")
|
| 47 |
+
return f"Error initializing agent: {e}", None
|
| 48 |
+
# 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)
|
| 49 |
+
agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
|
| 50 |
+
print(agent_code)
|
| 51 |
+
|
| 52 |
+
# 2. Fetch Questions
|
| 53 |
+
print(f"Fetching questions from: {questions_url}")
|
| 54 |
+
try:
|
| 55 |
+
response = requests.get(questions_url, timeout=15)
|
| 56 |
+
response.raise_for_status()
|
| 57 |
+
questions_data = response.json()
|
| 58 |
+
if not questions_data:
|
| 59 |
+
print("Fetched questions list is empty.")
|
| 60 |
+
return "Fetched questions list is empty or invalid format.", None
|
| 61 |
+
print(f"Fetched {len(questions_data)} questions.")
|
| 62 |
+
except requests.exceptions.RequestException as e:
|
| 63 |
+
print(f"Error fetching questions: {e}")
|
| 64 |
+
return f"Error fetching questions: {e}", None
|
| 65 |
+
except requests.exceptions.JSONDecodeError as e:
|
| 66 |
+
print(f"Error decoding JSON response from questions endpoint: {e}")
|
| 67 |
+
print(f"Response text: {response.text[:500]}")
|
| 68 |
+
return f"Error decoding server response for questions: {e}", None
|
| 69 |
+
except Exception as e:
|
| 70 |
+
print(f"An unexpected error occurred fetching questions: {e}")
|
| 71 |
+
return f"An unexpected error occurred fetching questions: {e}", None
|
| 72 |
+
|
| 73 |
+
# 3. Run your Agent
|
| 74 |
+
results_log = []
|
| 75 |
+
answers_payload = []
|
| 76 |
+
print(f"Running agent on {len(questions_data)} questions...")
|
| 77 |
+
for item in questions_data:
|
| 78 |
+
task_id = item.get("task_id")
|
| 79 |
+
question_text = item.get("question")
|
| 80 |
+
if not task_id or question_text is None:
|
| 81 |
+
print(f"Skipping item with missing task_id or question: {item}")
|
| 82 |
+
continue
|
| 83 |
+
try:
|
| 84 |
+
submitted_answer = agent(question_text)
|
| 85 |
+
answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
|
| 86 |
+
results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
|
| 87 |
+
except Exception as e:
|
| 88 |
+
print(f"Error running agent on task {task_id}: {e}")
|
| 89 |
+
results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": f"AGENT ERROR: {e}"})
|
| 90 |
+
|
| 91 |
+
if not answers_payload:
|
| 92 |
+
print("Agent did not produce any answers to submit.")
|
| 93 |
+
return "Agent did not produce any answers to submit.", pd.DataFrame(results_log)
|
| 94 |
+
|
| 95 |
+
# 4. Prepare Submission
|
| 96 |
+
submission_data = {"username": username.strip(), "agent_code": agent_code, "answers": answers_payload}
|
| 97 |
+
status_update = f"Agent finished. Submitting {len(answers_payload)} answers for user '{username}'..."
|
| 98 |
+
print(status_update)
|
| 99 |
+
|
| 100 |
+
# 5. Submit
|
| 101 |
+
print(f"Submitting {len(answers_payload)} answers to: {submit_url}")
|
| 102 |
+
try:
|
| 103 |
+
response = requests.post(submit_url, json=submission_data, timeout=60)
|
| 104 |
+
response.raise_for_status()
|
| 105 |
+
result_data = response.json()
|
| 106 |
+
final_status = (
|
| 107 |
+
f"Submission Successful!\n"
|
| 108 |
+
f"User: {result_data.get('username')}\n"
|
| 109 |
+
f"Overall Score: {result_data.get('score', 'N/A')}% "
|
| 110 |
+
f"({result_data.get('correct_count', '?')}/{result_data.get('total_attempted', '?')} correct)\n"
|
| 111 |
+
f"Message: {result_data.get('message', 'No message received.')}"
|
| 112 |
+
)
|
| 113 |
+
print("Submission successful.")
|
| 114 |
+
results_df = pd.DataFrame(results_log)
|
| 115 |
+
return final_status, results_df
|
| 116 |
+
except requests.exceptions.HTTPError as e:
|
| 117 |
+
error_detail = f"Server responded with status {e.response.status_code}."
|
| 118 |
+
try:
|
| 119 |
+
error_json = e.response.json()
|
| 120 |
+
error_detail += f" Detail: {error_json.get('detail', e.response.text)}"
|
| 121 |
+
except requests.exceptions.JSONDecodeError:
|
| 122 |
+
error_detail += f" Response: {e.response.text[:500]}"
|
| 123 |
+
status_message = f"Submission Failed: {error_detail}"
|
| 124 |
+
print(status_message)
|
| 125 |
+
results_df = pd.DataFrame(results_log)
|
| 126 |
+
return status_message, results_df
|
| 127 |
+
except requests.exceptions.Timeout:
|
| 128 |
+
status_message = "Submission Failed: The request timed out."
|
| 129 |
+
print(status_message)
|
| 130 |
+
results_df = pd.DataFrame(results_log)
|
| 131 |
+
return status_message, results_df
|
| 132 |
+
except requests.exceptions.RequestException as e:
|
| 133 |
+
status_message = f"Submission Failed: Network error - {e}"
|
| 134 |
+
print(status_message)
|
| 135 |
+
results_df = pd.DataFrame(results_log)
|
| 136 |
+
return status_message, results_df
|
| 137 |
+
except Exception as e:
|
| 138 |
+
status_message = f"An unexpected error occurred during submission: {e}"
|
| 139 |
+
print(status_message)
|
| 140 |
+
results_df = pd.DataFrame(results_log)
|
| 141 |
+
return status_message, results_df
|
| 142 |
+
|
| 143 |
+
|
| 144 |
+
# --- Build Gradio Interface using Blocks ---
|
| 145 |
+
with gr.Blocks() as demo:
|
| 146 |
+
gr.Markdown("# Basic Agent Evaluation Runner")
|
| 147 |
+
gr.Markdown(
|
| 148 |
+
"""
|
| 149 |
+
**Instructions:**
|
| 150 |
+
|
| 151 |
+
1. Please clone this space, then modify the code to define your agent's logic, the tools, the necessary packages, etc ...
|
| 152 |
+
2. Log in to your Hugging Face account using the button below. This uses your HF username for submission.
|
| 153 |
+
3. Click 'Run Evaluation & Submit All Answers' to fetch questions, run your agent, submit answers, and see the score.
|
| 154 |
+
|
| 155 |
+
---
|
| 156 |
+
**Disclaimers:**
|
| 157 |
+
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).
|
| 158 |
+
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.
|
| 159 |
+
"""
|
| 160 |
+
)
|
| 161 |
+
|
| 162 |
+
gr.LoginButton()
|
| 163 |
+
|
| 164 |
+
run_button = gr.Button("Run Evaluation & Submit All Answers")
|
| 165 |
+
|
| 166 |
+
status_output = gr.Textbox(label="Run Status / Submission Result", lines=5, interactive=False)
|
| 167 |
+
# Removed max_rows=10 from DataFrame constructor
|
| 168 |
+
results_table = gr.DataFrame(label="Questions and Agent Answers", wrap=True)
|
| 169 |
+
|
| 170 |
+
run_button.click(
|
| 171 |
+
fn=run_and_submit_all,
|
| 172 |
+
outputs=[status_output, results_table]
|
| 173 |
+
)
|
| 174 |
+
|
| 175 |
+
if __name__ == "__main__":
|
| 176 |
+
print("\n" + "-"*30 + " App Starting " + "-"*30)
|
| 177 |
+
# Check for SPACE_HOST and SPACE_ID at startup for information
|
| 178 |
+
space_host_startup = os.getenv("SPACE_HOST")
|
| 179 |
+
space_id_startup = os.getenv("SPACE_ID") # Get SPACE_ID at startup
|
| 180 |
+
|
| 181 |
+
if space_host_startup:
|
| 182 |
+
print(f"✅ SPACE_HOST found: {space_host_startup}")
|
| 183 |
+
print(f" Runtime URL should be: https://{space_host_startup}.hf.space")
|
| 184 |
+
else:
|
| 185 |
+
print("ℹ️ SPACE_HOST environment variable not found (running locally?).")
|
| 186 |
+
|
| 187 |
+
if space_id_startup: # Print repo URLs if SPACE_ID is found
|
| 188 |
+
print(f"✅ SPACE_ID found: {space_id_startup}")
|
| 189 |
+
print(f" Repo URL: https://huggingface.co/spaces/{space_id_startup}")
|
| 190 |
+
print(f" Repo Tree URL: https://huggingface.co/spaces/{space_id_startup}/tree/main")
|
| 191 |
+
else:
|
| 192 |
+
print("ℹ️ SPACE_ID environment variable not found (running locally?). Repo URL cannot be determined.")
|
| 193 |
+
|
| 194 |
+
print("-"*(60 + len(" App Starting ")) + "\n")
|
| 195 |
+
|
| 196 |
+
print("Launching Gradio Interface for Basic Agent Evaluation...")
|
| 197 |
demo.launch(debug=True, share=False)
|
requirements.txt
CHANGED
|
@@ -1,2 +1,13 @@
|
|
| 1 |
-
gradio
|
| 2 |
-
requests
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
gradio
|
| 2 |
+
requests
|
| 3 |
+
langchain==0.2.16
|
| 4 |
+
langchain-core==0.2.38
|
| 5 |
+
langchain-openai==0.1.22
|
| 6 |
+
google-generativeai
|
| 7 |
+
langchain-google-genai==1.0.10
|
| 8 |
+
langchain-experimental==0.0.65
|
| 9 |
+
langchain-community==0.2.16
|
| 10 |
+
langgraph==0.1.19
|
| 11 |
+
ddgs
|
| 12 |
+
duckduckgo-search==5.3.1
|
| 13 |
+
wikipedia
|