Update app.py

app.py

@@ -1,4 +1,4 @@
-# app_minimal.py - TEST VERSION
+# app_working.py
 import gradio as gr
 import os
 import json
@@ -6,50 +6,91 @@ from datasets import load_dataset
 from PIL import Image
 import io
 
-HF_DATASET_REPO = "alyex/karnak-data-app"
+HF_TOKEN = os.getenv("HF_TOKEN")
+DATASET_REPO = "alyex/karnak-data-app"
 
-with gr.Blocks(title="⚡ Minimal Test") as demo:
-    gr.Markdown("# 🧪 Testing Dataset Access")
+# Simple state
+current_index = 0
+manifest = []
+
+def load_test_manifest():
+    """Create a test manifest"""
+    return [
+        {"kiu_id": "00001", "instance_id": 0, "crop_coords": [0, 0, 100, 100]},
+        {"kiu_id": "00002", "instance_id": 0, "crop_coords": [0, 0, 100, 100]},
+    ]
+
+def get_current_data():
+    """Get data for current item"""
+    global current_index, manifest
     
-    kiu_input = gr.Textbox(label="Enter KIU ID", value="00001")
-    fetch_btn = gr.Button("Fetch Data")
+    if not manifest:
+        manifest = load_test_manifest()
     
-    image_output = gr.Image(label="Image")
-    html_output = gr.HTML(label="HTML Preview")
-    status = gr.Markdown()
+    if current_index >= len(manifest):
+        return None, "No items", "No data", ""
     
-    def fetch_data(kiu_id):
-        kiu_id = kiu_id.zfill(5)
+    item = manifest[current_index]
+    kiu_id = item["kiu_id"]
+    
+    # Try to fetch from dataset
+    try:
+        dataset = load_dataset(DATASET_REPO, streaming=True, token=HF_TOKEN)
         
-        try:
-            # Load dataset
-            dataset = load_dataset(HF_DATASET_REPO, streaming=True)
-            
-            # Find the KIU
-            found = False
-            html_content = ""
-            image_data = None
-            
-            for item in dataset["train"].take(100):  # Limit search
-                if item["kiu_id"] == kiu_id:
-                    found = True
-                    html_content = item.get("html", "No HTML")
-                    
-                    if item.get("image"):
-                        image = Image.open(io.BytesIO(item["image"]))
-                        image_data = image
-                    
-                    break
-            
-            if found:
-                html_display = f"<div style='padding:10px;background:#f0f0f0'>{html_content[:500]}...</div>"
-                return image_data, html_display, f"✅ Found KIU {kiu_id}"
-            else:
-                return None, "Not found", f"❌ KIU {kiu_id} not found in first 100 items"
+        for data_item in dataset["train"].take(100):
+            if data_item["kiu_id"] == kiu_id:
+                html = data_item.get("html", "No HTML")
+                image = None
+                
+                if data_item.get("image"):
+                    try:
+                        image = Image.open(io.BytesIO(data_item["image"]))
+                        # Apply crop
+                        x1, y1, x2, y2 = item["crop_coords"]
+                        image = image.crop((x1, y1, x2, y2))
+                    except:
+                        pass
                 
-        except Exception as e:
-            return None, f"Error: {str(e)}", f"❌ Failed: {str(e)}"
+                html_display = f"<div style='padding:10px'>{html[:1000]}...</div>"
+                info = f"KIU {kiu_id} | Instance {item['instance_id']}"
+                
+                return image, html_display, info, ""
+    
+    except Exception as e:
+        return None, f"Error: {str(e)}", f"KIU {kiu_id}", ""
+    
+    return None, "Not found in dataset", f"KIU {kiu_id}", ""
+
+with gr.Blocks(title="⚡ Working App") as demo:
+    gr.Markdown("# ⚡ Hieroglyph Annotation")
+    
+    image_display = gr.Image(label="Line Instance")
+    html_display = gr.HTML(label="Reference")
+    info_display = gr.Markdown()
+    line_input = gr.Textbox(label="Line Number")
+    
+    next_btn = gr.Button("Next")
+    prev_btn = gr.Button("Previous")
+    
+    def update_display():
+        return get_current_data()
+    
+    def next_item():
+        global current_index
+        if current_index < len(manifest) - 1:
+            current_index += 1
+        return get_current_data()
+    
+    def prev_item():
+        global current_index
+        if current_index > 0:
+            current_index -= 1
+        return get_current_data()
+    
+    # Initial load
+    demo.load(update_display, outputs=[image_display, html_display, info_display, line_input])
     
-    fetch_btn.click(fetch_data, inputs=[kiu_input], outputs=[image_output, html_output, status])
+    next_btn.click(next_item, outputs=[image_display, html_display, info_display, line_input])
+    prev_btn.click(prev_item, outputs=[image_display, html_display, info_display, line_input])
 
 demo.launch(server_name="0.0.0.0", server_port=7860)