Upload app.py

app.py

@@ -3,6 +3,8 @@ import os
 import requests
 
 from langchain_community.tools import DuckDuckGoSearchResults
+from langfuse import Langfuse, get_client
+from langfuse.langchain import CallbackHandler
 import pandas as pd
 
 from src.agent import Agent
@@ -15,6 +17,18 @@ tools = [
 ]
 tools.append(DuckDuckGoSearchResults())
 
+Langfuse(
+    public_key=os.environ.get("LANGFUSE_PUBLIC_KEY"),
+    secret_key=os.environ.get("LANGFUSE_SECRET_KEY"),
+    host='https://cloud.langfuse.com'
+)
+
+# Get the configured client instance
+langfuse = get_client()
+
+# Initialize the Langfuse handler
+langfuse_handler = CallbackHandler()
+
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -46,7 +60,8 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             model="gpt-4o",
             tools=tools,
             system_prompt_path="prompts/system_prompt.txt",
-            openai_api_key=os.environ.get("OPENAI_API_KEY")
+            openai_api_key=os.environ.get("OPENAI_API_KEY"),
+            langfuse_callback_handler=langfuse_handler
         )
     except Exception as e:
         print(f"Error instantiating agent: {e}")