Update app.py

app.py

@@ -9,7 +9,6 @@ import re
 from huggingface_hub import hf_hub_download
 import tempfile
 import numpy as np
-# IMPORTANT: Importing necessary types for MCP definitions
 from typing import List, Tuple, Dict, Any
 
 # --- 1. CONFIGURATION & SECRETS ---
@@ -190,11 +189,10 @@ def process_pipeline(image, conf_threshold):
         return None, f"System Failure: {str(e)}", "", None, str(e), None, []
 
 # --- 6. MCP API FUNCTIONS (Optimized for Claude) ---
-# These functions are explicitly defined as MCP tools at the bottom of the script.
-# They use TYPE HINTS so Claude knows what arguments to provide.
+# These functions use TYPE HINTS so Claude knows what arguments to provide.
 # They return FILE PATHS (strings) for images, not base64 objects.
 
-def detect_hieroglyphs(image: Image.Image, conf: float = 0.25) -> Tuple[str, Dict[str, Any]]:
+def detect_hieroglyphs_api(image: Image.Image, conf: float = 0.25) -> Tuple[str, Dict[str, Any]]:
     """
     Scans an image for Egyptian hieroglyphs.
     Returns a tuple containing:
@@ -211,7 +209,7 @@ def detect_hieroglyphs(image: Image.Image, conf: float = 0.25) -> Tuple[str, Dic
         
     return path, {"count": len(dets), "detections": dets}
 
-def translate_codes(keywords_text: str) -> Tuple[str, str]:
+def translate_codes_api(keywords_text: str) -> Tuple[str, str]:
     """
     Takes a comma-separated string of Gardiner codes (e.g., 'G43, X1, N5').
     Returns two translations: a mystical interpretation and an academic translation.
@@ -225,10 +223,10 @@ def translate_codes(keywords_text: str) -> Tuple[str, str]:
     clean_mystical = re.sub('<[^<]+?>', '', mystical) 
     return clean_mystical, academic
 
-def get_analytics_chart(json_data: Dict[str, Any]) -> str:
+def get_analytics_chart_api(json_data: Dict[str, Any]) -> str:
     """
     Generates statistical charts based on detection data.
-    Input: The JSON output from the 'detect_hieroglyphs' tool.
+    Input: The JSON output from the 'detect' tool.
     Returns: The file path to the generated chart image.
     """
     dets = json_data.get("detections", [])
@@ -245,7 +243,7 @@ def get_analytics_chart(json_data: Dict[str, Any]) -> str:
     plt.close(fig)
     return path
 
-def list_all_codes() -> Dict[str, Any]:
+def list_all_codes_api() -> Dict[str, Any]:
     """
     Returns the complete database of supported Gardiner codes and their descriptions.
     Useful for looking up specific symbol meanings.
@@ -449,7 +447,47 @@ with gr.Blocks(title="Rosetta Decoder Ultimate") as demo:
     gr.HTML(f"<style>{custom_css}</style>")
     gr.HTML(trail_script)
     
-    # (Removed hidden buttons - tools are now defined explicitly in launch())
+    # --- MCP TOOL REGISTRATION LAYER (Hidden from UI) ---
+    # This is the correct, standard way to register tools in Gradio for MCP.
+    # We use hidden buttons connected to our specific API functions.
+    # The 'api_name' argument defines the short name you see in Claude.
+    
+    with gr.Row(visible=False):
+        # 1. Detect Tool (Short name: "detect")
+        btn_detect = gr.Button("Detect")
+        btn_detect.click(
+            fn=detect_hieroglyphs_api,
+            inputs=[gr.Image(label="img"), gr.Number(label="conf")],
+            outputs=[gr.Textbox(label="path"), gr.JSON(label="json")],
+            api_name="detect" # << SHORT NAME HERE
+        )
+
+        # 2. Translate Tool (Short name: "translate")
+        btn_trans = gr.Button("Translate")
+        btn_trans.click(
+            fn=translate_codes_api,
+            inputs=[gr.Textbox(label="text")],
+            outputs=[gr.Textbox(label="mystic"), gr.Textbox(label="academic")],
+            api_name="translate" # << SHORT NAME HERE
+        )
+
+        # 3. Analytics Tool (Short name: "analytics")
+        btn_anal = gr.Button("Analytics")
+        btn_anal.click(
+            fn=get_analytics_chart_api,
+            inputs=[gr.JSON(label="data")],
+            outputs=[gr.Textbox(label="chart_path")],
+            api_name="analytics" # << SHORT NAME HERE
+        )
+
+        # 4. List Codes Tool (Short name: "list_codes")
+        btn_list = gr.Button("List")
+        btn_list.click(
+            fn=list_all_codes_api,
+            inputs=[],
+            outputs=[gr.JSON(label="data")],
+            api_name="list_codes" # << SHORT NAME HERE
+        )
 
     # --- Visible UI ---
     with gr.Row(elem_classes="header-row"):
@@ -509,12 +547,9 @@ with gr.Blocks(title="Rosetta Decoder Ultimate") as demo:
     btn_cam.click(fn=process_pipeline, inputs=[img_cam, slider_conf_cam], outputs=outputs)
 
 if __name__ == "__main__":
-    # FIX 2 & 3: Explicitly define short tool names and ensure images are handled correctly.
-    # By passing the functions directly to mcp_tools, Gradio uses the function name as the tool name.
-    # The /tmp/gradio_results path ensures Claude can read the generated image files.
+    # Standard launch. Tools are registered via the hidden buttons above.
     demo.launch(
         mcp_server=True, 
-        mcp_tools=[detect_hieroglyphs, translate_codes, get_analytics_chart, list_all_codes],
         ssr_mode=False, 
         allowed_paths=["/tmp", "/tmp/gradio_results", "."]
     )