fix import agent to app

agent.py

@@ -135,16 +135,6 @@ with open("system_prompt.txt", "r", encoding="utf-8") as f:
 sys_msg = SystemMessage(content=system_prompt)
 
 
-# generate the chat interface and tools
-llm = HuggingFaceEndpoint(
-    repo_id = "microsoft/Phi-4-reasoning-plus",
-    huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
-)
-
-
-chat = ChatHuggingFace(llm=llm, verbose=False)
-
-
 tools = [
     add,
     subtract,
@@ -157,42 +147,51 @@ tools = [
 ]
 
 
-chat_with_tools = chat.bind_tools(tools)
-
+# build graph function
+def build_graph():
+    # llm
+    llm = HuggingFaceEndpoint(
+        repo_id = "microsoft/Phi-4-reasoning-plus",
+        huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
+    )
 
-# generate AgentState and Agent graph
-class AgentState(TypedDict):
-    messages: Annotated[list[AnyMessage], add_messages]
+    chat = ChatHuggingFace(llm=llm, verbose=False)
 
-def assistant(state: AgentState):
-    return {
-        "messages": [chat_with_tools.invoke(state["messages"])],
-    }
+    # bind tools to llm
+    chat_with_tools = chat.bind_tools(tools)
 
+    # generate AgentState and Agent graph
+    class AgentState(TypedDict):
+        messages: Annotated[list[AnyMessage], add_messages]
 
-# build graph
-builder = StateGraph(AgentState)
+    def assistant(state: AgentState):
+        return {
+            "messages": [chat_with_tools.invoke(state["messages"])],
+        }
 
+    # build graph
+    builder = StateGraph(AgentState)
 
-# define nodes
-builder.add_node("assistant", assistant)
-builder.add_node("tools", ToolNode(tools))
+    # define nodes
+    builder.add_node("assistant", assistant)
+    builder.add_node("tools", ToolNode(tools))
 
+    # define edges
+    builder.add_edge(START, "assistant")
+    builder.add_conditional_edges(
+        "assistant",
+        # If the latest message requires a tool, route to tools
+        # Otherwise, provide a direct response
+        tools_condition,
+    )
+    builder.add_edge("tools", "assistant")
 
-# define edges
-builder.add_edge(START, "assistant")
-builder.add_conditional_edges(
-    "assistant",
-    # If the latest message requires a tool, route to tools
-    # Otherwise, provide a direct response
-    tools_condition,
-)
-builder.add_edge("tools", "assistant")
-graph = builder.compile()
+    return builder.compile()
 
 
 if __name__ == "__main__":
     question = "When was a picture of St. Thomas Aquinas first added to the Wikipedia page on the Principle of double effect?"
+    graph = build_graph()
     messages = [HumanMessage(content=question)]
     messages = graph.invoke({"messages": messages})
     for m in messages["messages"]:

app.py

@@ -4,6 +4,9 @@ import requests
 import inspect
 import pandas as pd
 
+from langchain_core.messages import HumanMessage
+from agent import build_graph
+
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -45,7 +48,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         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)
-    agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
+    agent_code = f"https://huggingface.co/spaces/errchh/hfagentscourse-finalassignment/tree/main"
     print(agent_code)
 
     # 2. Fetch Questions
@@ -91,7 +94,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         print("Agent did not produce any answers to submit.")
         return "Agent did not produce any answers to submit.", pd.DataFrame(results_log)
 
-    # 4. Prepare Submission 
+    # 4. Prepare Submission
     submission_data = {"username": username.strip(), "agent_code": agent_code, "answers": answers_payload}
     status_update = f"Agent finished. Submitting {len(answers_payload)} answers for user '{username}'..."
     print(status_update)
@@ -193,4 +196,4 @@ if __name__ == "__main__":
     print("-"*(60 + len(" App Starting ")) + "\n")
 
     print("Launching Gradio Interface for Basic Agent Evaluation...")
-    demo.launch(debug=True, share=False)
+    demo.launch(debug=True, share=False)