Update app.py

app.py

@@ -5,25 +5,30 @@ import os
 from huggingface_hub import hf_hub_download
 import numpy as np
 
-# --- 1. SETUP & MODEL LOADING (LOGIC UNCHANGED) ---
+# --- 1. SETUP & MODEL LOADING ---
 MODEL_REPO = "youkii-xr/hieroglyphic-detection"
 MODEL_FILENAME = "best.pt"
 
+# Fix for the Ultralytics config warning in containers
+os.environ["YOLO_CONFIG_DIR"] = "/tmp/Ultralytics"
+
 try:
+    print("System: Downloading model weights...")
     model_path = hf_hub_download(
         repo_id=MODEL_REPO,
         filename=MODEL_FILENAME,
         token=os.environ.get("HF_TOKEN")
     )
     model = YOLO(model_path)
+    print("System: Model loaded successfully.")
 except Exception as e:
     print(f"Error: {e}")
     model = None
 
-# --- 2. DETECTION LOGIC (LOGIC UNCHANGED) ---
+# --- 2. DETECTION LOGIC ---
 def detect_hieroglyphs(image: Image.Image, conf_threshold: float = 0.25):
     if image is None: return None, None
-    if model is None: return None, None
+    if model is None: return None, {"error": "Model failed to load"}
 
     try:
         results = model.predict(source=image, conf=conf_threshold, iou=0.45, imgsz=640, verbose=False, device='cpu', max_det=300)
@@ -57,7 +62,6 @@ def detect_hieroglyphs(image: Image.Image, conf_threshold: float = 0.25):
 
 # --- 3. UI/UX: THE EGYPTIAN NIGHT STYLE ---
 
-# This CSS mimics the VoiceKit structure but swaps Purple/Indigo for Lapis/Gold
 custom_css = """
 @import url('https://fonts.googleapis.com/css2?family=Cinzel:wght@400;700&display=swap');
 
@@ -71,7 +75,7 @@ custom_css = """
 
 body, .gradio-container {
     background: radial-gradient(circle at 50% 0%, #1a1f35 0%, #050510 100%) !important;
-    font-family: 'Cinzel', serif !important; /* The Ancient Font */
+    font-family: 'Cinzel', serif !important; 
 }
 
 /* --- THE GLASS CARDS --- */
@@ -87,7 +91,7 @@ body, .gradio-container {
 .card-title {
     font-size: 18px;
     font-weight: 700;
-    color: #d4af37; /* Gold */
+    color: #d4af37; 
     text-transform: uppercase;
     letter-spacing: 2px;
     margin-bottom: 20px;
@@ -123,11 +127,10 @@ body, .gradio-container {
     font-size: 13px;
     line-height: 1.6;
 }
-.json-key { color: #7aa2f7; } /* Blue */
-.json-string { color: #e0af68; } /* Gold-ish */
+.json-key { color: #7aa2f7; } 
+.json-string { color: #e0af68; } 
 
 /* --- BUTTONS --- */
-/* The "Action" Button */
 button.primary-btn {
     background: linear-gradient(135deg, #b8860b 0%, #d4af37 100%) !important;
     border: 1px solid #ffd700 !important;
@@ -144,8 +147,7 @@ button.primary-btn:hover {
     box-shadow: 0 0 25px rgba(212, 175, 55, 0.6);
 }
 
-/* --- INPUTS & OUTPUTS --- */
-/* Transparent backgrounds for images/json to blend into cards */
+/* --- ELEMENT VISIBILITY --- */
 .gradio-image, .gradio-json {
     background: transparent !important;
     border: none !important;
@@ -172,7 +174,6 @@ header_html = """
 </div>
 """
 
-# HTML for the Claude Config "Terminal"
 terminal_html = """
 <div class="card">
     <div class="card-title">⚡ CLAUDE DESKTOP CONFIG</div>
@@ -196,51 +197,49 @@ terminal_html = """
 """
 
 # --- 4. APP ASSEMBLY ---
-with gr.Blocks(css=custom_css, title="Horus Vision") as demo:
+# FIX: Removed 'css' argument here. We inject it via gr.HTML below.
+with gr.Blocks(title="Horus Vision") as demo:
+    
+    # FIX: Inject CSS directly into the HTML. This works on ALL Gradio versions.
+    gr.HTML(f"<style>{custom_css}</style>")
     
     # 1. Header Injection
     gr.HTML(header_html)
     
     # 2. Top Row: Info & Config
     with gr.Row():
-        # Left: The Model Info Card
         with gr.Column(scale=1):
             gr.HTML("""
             <div class="card" style="height: 100%;">
                 <div class="card-title">📜 MISSION BRIEF</div>
                 <p style="color: #ccc; line-height: 1.6;">
-                    This system utilizes the YOLOv8 architecture to detect and classify Gardiner codes from images of papyrus or stone.
+                    This system utilizes the YOLOv8 architecture to detect and classify Gardiner codes.
                     <br><br>
                     <span style="color: #d4af37;">> Model:</span> YOLOv8 Custom<br>
-                    <span style="color: #d4af37;">> Weights:</span> Private (Secure)<br>
                     <span style="color: #d4af37;">> Security:</span> MCP Standard
                 </p>
             </div>
             """)
             
-        # Right: The Terminal Card
         with gr.Column(scale=1):
             gr.HTML(terminal_html)
 
-    # 3. Main Detector Area (The "Workstation")
-    gr.HTML("<br>") # Spacer
+    # 3. Main Detector Area
+    gr.HTML("<br>") 
     
     with gr.Row():
         with gr.Column():
-            # We wrap the input area in a HTML card wrapper logic
             gr.HTML('<div class="card"> <div class="card-title">👁️ INPUT SOURCE</div>')
             img_input = gr.Image(type="pil", label="Upload Papyrus", sources=["upload", "clipboard", "webcam"], elem_classes="gradio-image")
             conf_slider = gr.Slider(0.1, 1.0, 0.25, step=0.05, label="Detection Confidence")
-            
-            # The Golden Button
             analyze_btn = gr.Button("🔮 DECIPHER SYMBOLS", elem_classes="primary-btn")
-            gr.HTML('</div>') # Close card
+            gr.HTML('</div>') 
 
         with gr.Column():
             gr.HTML('<div class="card"> <div class="card-title">💎 DECODED ARTIFACT</div>')
             img_output = gr.Image(label="Annotated Result", interactive=False, elem_classes="gradio-image")
             json_output = gr.JSON(label="Glyph Data", elem_classes="gradio-json")
-            gr.HTML('</div>') # Close card
+            gr.HTML('</div>') 
 
     # 4. Wiring
     analyze_btn.click(