Update app.py

app.py

@@ -77,7 +77,7 @@ def load_methods_from_tools():
 method_mapping = load_methods_from_tools()
 
 # Main request router method
-def request_router(name: str, input_data: Dict[str, Any]) -> Dict[str, Any]:
+def request_routerOld(name: str, input_data: Dict[str, Any]) -> Dict[str, Any]:
     # Reload methods to include newly created tools
     global method_mapping
     method_mapping = load_methods_from_tools()
@@ -93,7 +93,26 @@ def request_router(name: str, input_data: Dict[str, Any]) -> Dict[str, Any]:
         return output
     except Exception as e:
         return {"error": str(e)}
+        
+def request_router(name: str, input_data: Dict[str, Any]) -> Dict[str, Any]:
+    global method_mapping
+    method_mapping = load_methods_from_tools()
 
+    logging.info(f"Request to invoke method: {name} with input: {input_data}")
+    method = method_mapping.get(name)
+    if method is None:
+        error_message = "Method not found"
+        logging.error(error_message)
+        return {"error": error_message}
+    
+    try:
+        output = method(input_data)
+        logging.info(f"Output: {output}")
+        return output
+    except Exception as e:
+        logging.error(f"Error occurred: {str(e)}")
+        return {"error": str(e)}
+        
 # Gradio Interface
 def launch_gradio_app():
     with gr.Blocks() as demo:
@@ -112,5 +131,22 @@ def launch_gradio_app():
 
     demo.launch()
 
+import logging
+
+# Set up logging configuration
+logging.basicConfig(level=logging.INFO, filename='request_logs.txt', 
+                    format='%(asctime)s - %(message)s')
+
+def log_payload(*args, **kwargs):
+    # Log the incoming payload
+    logging.info(f"Received payload: args={args}, kwargs={kwargs}")
+    return main_function(*args, **kwargs)
+
+def main_function(name):
+    return "Hello " + name + "!!"
+
+
+
+
 if __name__ == "__main__":
     launch_gradio_app()