Update app.py

app.py

@@ -1,33 +1,51 @@
 # filename: request_router.py
 
 import gradio as gr
+import os
+import importlib.util
 from typing import Any, Dict
 
-# Example methods to demonstrate routing
-def method_one(input_data: Dict[str, Any]) -> Dict[str, Any]:
-    # Process input_data for method_one
-    return {"result": "Method One executed", "input": input_data}
+# Define the path to the tools folder
+TOOLS_DIR = './tools'
 
-def method_two(input_data: Dict[str, Any]) -> Dict[str, Any]:
-    # Process input_data for method_two
-    return {"result": "Method Two executed", "input": input_data}
+# Function to dynamically load methods from files in the ./tools folder
+def load_methods_from_tools():
+    method_mapping = {}
+    
+    # List all .py files in the tools directory
+    for filename in os.listdir(TOOLS_DIR):
+        if filename.endswith('.py'):
+            method_name = filename[:-3].lower()  # Remove '.py' and convert to lowercase
+            
+            # Load the module
+            module_path = os.path.join(TOOLS_DIR, filename)
+            spec = importlib.util.spec_from_file_location(method_name, module_path)
+            module = importlib.util.module_from_spec(spec)
+            spec.loader.exec_module(module)
+
+            # Map all functions from the module to the method mapping
+            for attr in dir(module):
+                if callable(getattr(module, attr)) and not attr.startswith("__"):
+                    method_mapping[method_name] = getattr(module, attr)
+
+    return method_mapping
+
+# Load the method mapping at the start
+method_mapping = load_methods_from_tools()
 
 # Main request router method
 def request_router(name: str, input_data: Dict[str, Any]) -> Dict[str, Any]:
-    # Mapping method names to actual functions
-    method_mapping = {
-        "method_one": method_one,
-        "method_two": method_two
-    }
-
     # Retrieve the appropriate method based on the name
     method = method_mapping.get(name)
     if method is None:
         return {"error": "Method not found"}
-
+    
     # Call the method with the provided input data
-    output = method(input_data)
-    return output
+    try:
+        output = method(input_data)
+        return output
+    except Exception as e:
+        return {"error": str(e)}
 
 # Gradio Interface
 def launch_gradio_app():