Update app.py

app.py

@@ -1,7 +1,22 @@
 import yaml
 import os
 from smolagents import GradioUI, CodeAgent, InferenceClientModel
-from smolagents.monitoring import LangfuseCallback
+
+# Langfuse + OpenTelemetry setup
+from langfuse import get_client
+from openinference.instrumentation.smolagents import SmolagentsInstrumentor
+
+# Initialize Langfuse client (reads from environment variables automatically)
+langfuse = get_client()
+
+# Verify connection
+if langfuse.auth_check():
+    print("✅ Langfuse client authenticated!")
+else:
+    print("❌ Langfuse authentication failed!")
+
+# Initialize OpenTelemetry instrumentation
+SmolagentsInstrumentor().instrument()
 
 # Get current directory path
 CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
@@ -13,13 +28,6 @@ from tools.catering_service_tool import SimpleTool as CateringServiceTool
 from tools.superhero_party_theme_generator import SuperheroPartyThemeTool as SuperheroPartyThemeGenerator
 from tools.final_answer import FinalAnswerTool as FinalAnswer
 
-# Initialize Langfuse callback (reads from HF Spaces secrets) - @mangoobee
-langfuse_callback = LangfuseCallback(
-    public_key=os.environ["LANGFUSE_PUBLIC_KEY"],
-    secret_key=os.environ["LANGFUSE_SECRET_KEY"],
-    host=os.environ["LANGFUSE_HOST"]
-)
-
 model = InferenceClientModel(
     model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
     provider=None,
@@ -44,9 +52,9 @@ agent = CodeAgent(
     planning_interval=None,
     name=None,
     description=None,
-    prompt_templates=prompt_templates,
-    callbacks=[langfuse_callback]  # Add this parameter
+    prompt_templates=prompt_templates
+    # No callbacks parameter needed - instrumentation happens globally
 )
 
 if __name__ == "__main__":
-    GradioUI(agent).launch()
+    GradioUI(agent).launch()