Update app.py

app.py

@@ -696,6 +696,24 @@ class MovieKnowledgeBase:
 # Initialize the knowledge base with embeddings
 movie_kb = MovieKnowledgeBase(movie_knowledge_base)
 
+# Function tools with logging
+def log_function_tool(func):
+    """Decorator to log function tool calls"""
+    @functools.wraps(func)
+    def wrapper(*args, **kwargs):
+        # Get function inputs
+        func_name = func.__name__
+        func_inputs = kwargs if kwargs else args[0] if args else {}
+        
+        # Run the function
+        result = func(*args, **kwargs)
+        
+        # Log function call
+        logger.log_function_call(func_name, func_inputs, result)
+        
+        return result
+    return wrapper
+
 def find_similar_movies(query_description, top_n=3):
     """
     Find movies similar to the query description using sentence embeddings and cosine similarity.