Update app.py

app.py

@@ -4,6 +4,7 @@ import pickle
 import numpy as np
 import os
 import random
+import base64
 from huggingface_hub import InferenceClient
 from sklearn.metrics.pairwise import cosine_similarity
 from IO_pipeline import RecipeDigitalizerPipeline 
@@ -24,7 +25,6 @@ try:
         if isinstance(data, dict):
             stored_embeddings = np.array(data['embeddings'])
         elif isinstance(data, pd.DataFrame):
-            # Fallback for dataframe format
             target_col = next((c for c in ['embedding', 'embeddings', 'vectors'] if c in data.columns), None)
             stored_embeddings = np.vstack(data[target_col].values) if target_col else data
         else:
@@ -36,7 +36,19 @@ except Exception as e:
     stored_embeddings = None
 
 # ==========================================
-# 2. LOGIC
+# 2. HELPER: IMAGE TO BASE64 (The Fix)
+# ==========================================
+def image_to_base64(image_path):
+    if not os.path.exists(image_path):
+        return ""
+    with open(image_path, "rb") as img_file:
+        return base64.b64encode(img_file.read()).decode('utf-8')
+
+# Pre-load the logo
+logo_b64 = image_to_base64("logo.jpg")
+
+# ==========================================
+# 3. BACKEND LOGIC
 # ==========================================
 def get_embedding_via_api(text):
     if not client: raise ValueError("HF_TOKEN missing")
@@ -79,7 +91,7 @@ def magic_pipeline(image_path):
         return f"Error: {e}", f"Error: {e}"
 
 # ==========================================
-# 3. CSS STYLING (With Logo Support)
+# 4. CSS STYLING
 # ==========================================
 fb_css = """
 body {background-color: #f0f2f5 !important; font-family: Helvetica, Arial, sans-serif;}
@@ -136,15 +148,15 @@ body {background-color: #f0f2f5 !important; font-family: Helvetica, Arial, sans-
 """
 
 # ==========================================
-# 4. LAYOUT
+# 5. LAYOUT
 # ==========================================
 with gr.Blocks(css=fb_css, title="Legacy Kitchen") as demo:
     
- # --- HEADER WITH LOGO ---
-    # Note the "/" before file/logo.jpg
-    gr.HTML("""
+    # --- HEADER WITH EMBEDDED LOGO ---
+    # We inject the Base64 string directly into the HTML
+    gr.HTML(f"""
         <div class="fb-header">
-            <img src="/file/logo.jpg" class="logo-img" alt="Logo">
+            <img src="data:image/jpeg;base64,{logo_b64}" class="logo-img" alt="Logo">
             Legacy Kitchen
         </div>
     """)
@@ -224,5 +236,4 @@ with gr.Blocks(css=fb_css, title="Legacy Kitchen") as demo:
     nav_saved.click(lambda: (gr.update(visible=False), gr.update(visible=False), gr.update(visible=True)), None, [feed_view, digitalizer_view, saved_view])
 
 if __name__ == "__main__":
-
-    demo.launch(allowed_paths=["."])
+    demo.launch()