Update app.py

app.py

@@ -98,7 +98,7 @@ except Exception as e:
     print(f"Error loading model: {e}")
     model = None
 
-# --- 5. CORE LOGIC ---
+# --- 5. CORE LOGIC (Helpers) ---
 
 def clean_ai_text(text):
     text = re.sub(r'^\d+[\.\)]\s*', '', text)
@@ -146,28 +146,61 @@ def generate_analytics_plots(detections, img_width, img_height):
     plt.tight_layout(pad=4.0)
     return fig
 
-def get_ai_translation_all_styles(keywords_list):
-    if not GOOGLE_API_KEY:
-        return "⚠️ Error: Google API Key not found in Secrets.", "Error"
-    if not keywords_list:
-        return "No hieroglyphs detected.", "No data."
-
-    keywords_str = ", ".join(keywords_list)
-    prompt = f"""
-    You are an expert Egyptologist AI. I have detected these symbols: [{keywords_str}].
-    Please provide 2 distinct outputs separated by "|||SEPARATOR|||".
-    1. A Mystical Story: Highly atmospheric, sounding like an ancient prophecy. 
-       Do NOT number this section. Use HTML tags <b> for bolding keywords and <br> for new lines.
-    2. An Academic Translation: Direct, linguistic, focusing on grammar. Use standard text.
-    """
+# --- 6. MCP EXPOSED FUNCTIONS ---
+
+@gr.tool
+def detect_hieroglyphs_mcp(image_path: str, conf_threshold: float = 0.25):
+    """Detects hieroglyphs in an image file. Returns JSON data."""
+    if model is None: return {"error": "Model not loaded"}
+    try:
+        image = Image.open(image_path)
+        img_w, img_h = image.size
+        results = model.predict(source=image, conf=conf_threshold, iou=0.45, imgsz=640, verbose=False, device='cpu', max_det=300)
+        detections = []
+        unique_codes = set()
+        for box in results[0].boxes:
+            if box.cls.numel() > 0:
+                cls_id = int(box.cls[0])
+                if 0 <= cls_id < len(model.names):
+                    code = model.names[cls_id]
+                    conf = float(box.conf[0])
+                    unique_codes.add(code)
+                    xyxy = box.xyxy[0].tolist()
+                    meaning = gardiner_map.get(code, "Unknown")
+                    detections.append({"code": code, "meaning": meaning, "confidence": round(conf, 2), "box": xyxy})
+        return {"status": "success", "image_size": [img_w, img_h], "total_detected": len(detections), "unique_codes": list(unique_codes), "detections": detections}
+    except Exception as e: return {"error": str(e)}
+
+@gr.tool
+def translate_hieroglyphs_mcp(keywords: list[str], style: str = "Academic Literal"):
+    """Translates a list of meanings using Gemini."""
+    if not GOOGLE_API_KEY: return "Error: Google API Key not configured."
+    keywords_str = ", ".join(keywords)
+    prompts = {
+        "Mystical Story": f"Create a coherent, mystical, and atmospheric short story using these ancient Egyptian concepts: [{keywords_str}]. Make it sound like a prophecy.",
+        "Academic Literal": f"Provide a direct, grammatical translation of these concepts: [{keywords_str}]. Focus on linguistic structure and sentence formation used in Egyptology.",
+        "Modern Interpretation": f"Translate the essence of these symbols: [{keywords_str}] into a modern, relatable piece of advice or horoscopic reading."
+    }
+    prompt = prompts.get(style, prompts["Academic Literal"])
     try:
         client = genai.Client(api_key=GOOGLE_API_KEY)
-        response = client.models.generate_content(model="gemini-2.5-flash", contents=prompt)
-        parts = response.text.split("|||SEPARATOR|||")
-        if len(parts) < 2: return clean_ai_text(response.text), "Could not parse academic style."
-        return clean_ai_text(parts[0].strip()), clean_ai_text(parts[1].strip())
-    except Exception as e:
-        return f"Error: {str(e)}", "Error"
+        response = client.models.generate_content(model="gemini-2.5-flash", contents=f"You are an expert Egyptologist AI. {prompt}")
+        return response.text
+    except Exception as e: return f"Translation Error: {str(e)}"
+
+@gr.tool
+def get_gardiner_info_mcp(code: str):
+    """Retrieves detailed info about a specific code."""
+    info = gardiner_data.get(code)
+    if info: return info
+    return {"error": f"Code {code} not found in database."}
+
+@gr.tool
+def get_supported_codes_mcp():
+    """Returns list of supported codes."""
+    return list(gardiner_data.keys())
+
+# --- 7. UI PROCESSING PIPELINE ---
 
 def process_pipeline(image, conf_threshold):
     if image is None: return None, "", "", None, "", None, []
@@ -196,10 +229,30 @@ def process_pipeline(image, conf_threshold):
                     crops.append((crop_img, f"{code}\n({int(conf*100)}%)"))
 
         mapped_words = [gardiner_map.get(code, f"[{code}]") for code in unique_codes]
+        
+        # Translation
+        if not GOOGLE_API_KEY:
+            mystical, academic = "API Key Missing", "API Key Missing"
+        else:
+            keywords_str = ", ".join(mapped_words)
+            prompt = f"""
+            You are an expert Egyptologist AI. I have detected these symbols: [{keywords_str}].
+            Please provide 2 distinct outputs separated by "|||SEPARATOR|||".
+            1. A Mystical Story: Highly atmospheric, sounding like an ancient prophecy. 
+               Do NOT number this section. Use HTML tags <b> for bolding keywords and <br> for new lines.
+            2. An Academic Translation: Direct, linguistic, focusing on grammar. Use standard text.
+            """
+            try:
+                client = genai.Client(api_key=GOOGLE_API_KEY)
+                response = client.models.generate_content(model="gemini-2.5-flash", contents=prompt)
+                parts = response.text.split("|||SEPARATOR|||")
+                if len(parts) < 2: mystical, academic = clean_ai_text(response.text), "Could not parse academic style."
+                else: mystical, academic = clean_ai_text(parts[0].strip()), clean_ai_text(parts[1].strip())
+            except Exception as e:
+                mystical, academic = f"Error: {str(e)}", f"Error: {str(e)}"
+
         analytics_plot = generate_analytics_plots(detections, img_w, img_h)
-        mystical, academic = get_ai_translation_all_styles(mapped_words)
         text_report = f"Total Symbols: {len(detections)}\nUnique Codes: {', '.join(unique_codes)}"
-        
         formatted_mystical = f"""<div class="mystical-container"><h3>✨ THE ANCIENT WHISPER</h3><p>{mystical}</p></div>"""
         json_output = {"count": len(detections), "detections": detections}
 
@@ -208,7 +261,7 @@ def process_pipeline(image, conf_threshold):
     except Exception as e:
         return None, f"System Failure: {str(e)}", "", None, str(e), None, []
 
-# --- 6. UI STYLING & ASSETS ---
+# --- 8. UI STYLING & ASSETS ---
 
 cursor_url = "url('data:image/svg+xml;base64,PHN2ZyB4bWxucz0iaHR0cDovL3d3dy53My5vcmcvMjAwMC9zdmciIHdpZHRoPSIzMiIgaGVpZ2h0PSIzMiIgdmlld0JveD0iMCAwIDMyIDMyIj4KICA8ZyBmaWxsPSJub25lIiBzdHJva2U9IiNkNGFmMzciIHN0cm9rZS13aWR0aD0iMS41Ij4KICAgIDxwYXRoIGQ9Ik0xNiw4IEM2LDIwIDI2LDIwIDE2LDggWiIgZmlsbD0icmdiYSgyMTIsIDE3NSwgNTUsIDAuMSkiLz4KICAgIDxjaXJjbGUgY3g9IjE2IiBjeT0iMTUiIHI9IjMiIGZpbGw9IiNkNGFmMzciLz4KICAgIDxwYXRoIGQ9Ik0xNiwyMiBMMTYsMjggTDEwLDI4Ii8+CiAgPC9nPgo8L3N2Zz4=')"
 
@@ -279,18 +332,18 @@ code {{ font-family: 'Space Mono', monospace; background: rgba(0,0,0,0.3); paddi
 button.primary-btn {{ background: var(--btn-grad) !important; border: 1px solid var(--border-color) !important; color: #000 !important; font-weight: 700 !important; font-size: 16px !important; }}
 .gradio-image, .gradio-json {{ background: transparent !important; border: none !important; }}
 
-/* UPDATED TOGGLE BUTTON CSS */
-button.toggle-btn {
-    background: #0a0a2e !important; /* Matches dark blue background */
-    border: 1px solid var(--border-color) !important;
-    color: var(--text-accent) !important;
-    padding: 5px 15px !important;
-    font-family: 'Space Mono', monospace;
+/* FIXED TOGGLE BUTTON */
+button.toggle-btn {{
+    background: #0a0a2e !important; 
+    border: 1px solid var(--border-color) !important; 
+    color: var(--text-accent) !important; 
+    padding: 5px 15px !important; 
+    font-family: 'Space Mono', monospace; 
     box-shadow: none !important;
-}
-button.toggle-btn:hover {
+}}
+button.toggle-btn:hover {{
     background: var(--info-bg) !important;
-}
+}}
 """
 
 header_html = """
@@ -326,34 +379,11 @@ guide_html = """
 </div>
 """
 
-claude_json_content = """{
-  "mcpServers": {
-    "gradio": {
-      "command": "npx",
-      "args": [
-        "mcp-remote",
-        "https://youkii-xr-hieroglyph-mcp-server.hf.space/gradio_api/mcp/",
-        "--transport",
-        "streamable-http"
-      ]
-    },
-    "upload_helper": {
-      "command": "C:\\\\Python313\\\\python.exe",
-      "args": [
-        "-m",
-        "gradio",
-        "upload-mcp",
-        "https://youkii-xr-hieroglyph-mcp-server.hf.space/",
-        "C:\\\\Claude_Work"
-      ]
-    }
-  }
-}"""
+claude_json_content = """{ "mcpServers": { "gradio": { "command": "npx", "args": [ "mcp-remote", "https://youkii-xr-hieroglyph-mcp-server.hf.space/gradio_api/mcp/", "--transport", "streamable-http" ] }, "upload_helper": { "command": "C:\\\\Python313\\\\python.exe", "args": [ "-m", "gradio", "upload-mcp", "https://youkii-xr-hieroglyph-mcp-server.hf.space/", "C:\\\\Claude_Work" ] } } }"""
 
-# JS for Gold Dust Trail
 trail_script = """<script>document.addEventListener('DOMContentLoaded', () => { document.addEventListener('mousemove', (e) => { if (Math.random() > 0.7) return; const dust = document.createElement('div'); dust.classList.add('gold-dust'); dust.style.left = e.clientX + 'px'; dust.style.top = e.clientY + 'px'; document.body.appendChild(dust); setTimeout(() => dust.remove(), 600); }); });</script>"""
 
-# --- 7. MAIN APP ASSEMBLY ---
+# --- 8. MAIN APP ASSEMBLY ---
 
 with gr.Blocks(title="Rosetta Decoder Ultimate") as demo:
     gr.HTML(f"<style>{custom_css}</style>")
@@ -397,13 +427,12 @@ with gr.Blocks(title="Rosetta Decoder Ultimate") as demo:
         # TAB 2: ABOUT
         with gr.TabItem("📜 VISION & ABOUT"): gr.HTML(mission_html)
 
-        # TAB 3: SETUP (Reordered)
+        # TAB 3: SETUP
         with gr.TabItem("🤖 SYSTEM SETUP"):
             gr.HTML(guide_html)
-            # Updated to show 25 lines so code box is expanded
             gr.Code(value=claude_json_content, language="json", label="claude_desktop_config.json", interactive=False, lines=25, container=True)
 
-        # TAB 4: GARDINER CODES (Reordered)
+        # TAB 4: GARDINER CODES
         with gr.TabItem("𓀀 GARDINER CODES"):
             gr.HTML(f"""<div class="card"><div class="card-title">📜 SUPPORTED GARDINER CODES</div><div style="overflow-x: auto; max-height: 600px; overflow-y: auto;"><table class="gardiner-table"><thead><tr><th>Code</th><th>Description</th><th>Transliteration</th><th>Type</th></tr></thead><tbody>{GARDINER_TABLE_CONTENT}</tbody></table></div></div>""")