Update app.py

app.py

@@ -15,7 +15,7 @@ if os.path.exists('images.zip'):
     with zipfile.ZipFile('images.zip', 'r') as zip_ref:
         zip_ref.extractall(IMAGE_DIR)
 
-# --- 2. טעינה וסנכרון ברזל (העוגן של הדאטה) ---
+# --- 2. טעינה וסנכרון ברזל (מבטיח שהמידע מה-CSV יהיה מדויק) ---
 model = SentenceTransformer('all-MiniLM-L6-v2')
 
 def load_and_sync_data():
@@ -26,7 +26,7 @@ def load_and_sync_data():
         user_emb = pickle.load(f)
         if isinstance(user_emb, list): user_emb = np.array(user_emb)
     
-    # חיתוך אחיד למניעת הסטה - המבורגר נשאר המבורגר
+    # חיתוך אחיד למניעת הסטה - מבטיח המבורגר = המבורגר
     min_l = min(len(df), len(dish_emb), len(user_emb))
     df = df.iloc[:min_l].reset_index(drop=True)
     dish_emb = dish_emb[:min_l]
@@ -34,11 +34,9 @@ def load_and_sync_data():
     return df, dish_emb, user_emb
 
 main_df, dish_embeddings, user_embeddings = load_and_sync_data()
-
-# שמות אמריקאיים לגיבוי ממליצים
 AMERICAN_NAMES = ["James Miller", "Sarah Johnson", "Michael Brown", "Emily Davis", "Robert Wilson", "Sophia Martinez"]
 
-# --- 3. מנוע החיפוש (Discovery) ---
+# --- 3. מנוע החיפוש (Discovery) עם כל הפיצ'רים מהדאטה ---
 def run_discovery(query, origin, hobbies, style):
     alpha = 0.7 
     q_vec = model.encode([str(query)])
@@ -53,15 +51,14 @@ def run_discovery(query, origin, hobbies, style):
     res = main_df.copy()
     res['similarity_score'] = final_scores
     
-    # סינון איכות (4.0+) ומיון
-    res['rating'] = pd.to_numeric(res['rating'], errors='coerce').fillna(0)
-    top_candidates = res[res['rating'] >= 4.0].sort_values('similarity_score', ascending=False)
+    # סינון ומיון (לוקחים טופ 10 כדי לוודא 3 תאומים שונים)
+    top_candidates = res.sort_values('similarity_score', ascending=False).head(10)
 
     final_results = []
     seen_users = set()
     
     for idx, row in top_candidates.iterrows():
-        # בחירת שם ממליץ
+        # שם הממליץ (מהדאטה או רנדומלי)
         u_name = row['user_name']
         if u_name == "N/A" or not u_name:
             u_name = random.choice(AMERICAN_NAMES)
@@ -75,11 +72,12 @@ def run_discovery(query, origin, hobbies, style):
     for original_idx, row, u_name in final_results:
         pct = f"{min(99.0, 85 + (row['similarity_score'] * 15)):.1f}%"
         
-        # חיפוש תמונה בתוך Dishes_Images בפורמט dish_{idx}
+        # חיפוש תמונה - סורק את כל התיקיות בתוך הזיפ שחולץ
         pattern = os.path.join(IMAGE_DIR, "**", f"dish_{original_idx}_*.jpg")
         img_files = glob.glob(pattern, recursive=True)
         img_url = f"file/{img_files[0]}" if img_files else "https://via.placeholder.com/400x400?text=BiteWise+Dish"
         
+        # תצוגת הכרטיסייה עם כל הנתונים מה-CSV (מחיר, עם מי לאכול, סוג מטבח)
         html_output += f"""
         <div style="border: 1px solid #C4A484; border-radius: 4px; padding: 25px; margin-bottom: 30px; background: #FFF9F5; box-shadow: 5px 5px 15px rgba(0,0,0,0.05); border-left: 10px solid #3E2723; display: flex; gap: 25px;">
             <img src="{img_url}" style="width: 250px; height: 250px; object-fit: cover; border: 4px solid #FAF9F6; outline: 1px solid #D2B48C;">
@@ -88,8 +86,15 @@ def run_discovery(query, origin, hobbies, style):
                     <h2 style="margin: 0; color: #3E2723; font-family: 'Playfair Display', serif; font-size: 2.2em;">{row['dish_name']}</h2>
                     <span style="background: #3E2723; color: white; padding: 2px 12px; font-weight: bold; border-radius: 20px; font-size: 0.85em;">{pct} MATCH</span>
                 </div>
-                <p style="color: #7F4F24; margin: 8px 0; font-family: 'Courier New', monospace; font-weight: bold;">📍 {row['restaurant_name']}</p>
+                <p style="color: #7F4F24; margin: 8px 0; font-family: 'Courier New', monospace; font-weight: bold;">📍 {row['restaurant_name']} | {row.get('cuisine_type', 'Gourmet')}</p>
                 <p style="font-family: 'Playfair Display', serif; font-style: italic; color: #4E342E; margin: 15px 0; font-size: 1.2em; line-height: 1.4;">"{row['taste_review']}"</p>
+                
+                <div style="display: flex; gap: 10px; margin-bottom: 15px; flex-wrap: wrap;">
+                    <span style="background: #EEDDCC; color: #3E2723; padding: 4px 10px; font-size: 0.75em; border: 1px solid #D2B48C; font-family: 'Courier New', monospace;">💰 PRICE: {row.get('price_range', '$$')}</span>
+                    <span style="background: #EEDDCC; color: #3E2723; padding: 4px 10px; font-size: 0.75em; border: 1px solid #D2B48C; font-family: 'Courier New', monospace;">👥 BEST FOR: {row.get('best_for', 'Friends')}</span>
+                    <span style="background: #EEDDCC; color: #3E2723; padding: 4px 10px; font-size: 0.75em; border: 1px solid #D2B48C; font-family: 'Courier New', monospace;">✨ VIBE: {row.get('food_vibe', 'Cozy')}</span>
+                </div>
+
                 <div style="background: #F5F1EE; padding: 12px; border-radius: 4px; font-size: 0.9em; color: #5D4037;">
                     <b>Twin:</b> {u_name} from {row['user_origin']} | <b>Rating:</b> {row['rating']}/5.0
                 </div>
@@ -98,7 +103,10 @@ def run_discovery(query, origin, hobbies, style):
         """
     return html_output
 
-# --- 4. הממשק ---
+def archive_submission():
+    return "### Thank you for your review! \nYour discovery has been successfully synced and shared with the BiteWise archive."
+
+# --- 4. ממשק בעיצוב Vintage Soul ---
 custom_css = """
 @import url('https://fonts.googleapis.com/css2?family=Playfair+Display:ital,wght@0,700;1,400&display=swap');
 .gradio-container { background-color: #FDFCF8 !important; color: #3E2723; }
@@ -128,9 +136,12 @@ with gr.Blocks(css=custom_css) as demo:
 
         with gr.Tab("03. ARCHIVE", id=2):
             gr.Markdown("### ARCHIVE A NEW CULINARY DISCOVERY")
-            gr.Textbox(label="DISH")
-            gr.Textbox(label="ESTABLISHMENT")
+            with gr.Row():
+                gr.Textbox(label="DISH")
+                gr.Textbox(label="ESTABLISHMENT")
             gr.Textbox(label="REVIEW", lines=3)
-            gr.Button("SUBMIT TO ARCHIVE", variant="primary")
+            submit_btn = gr.Button("SUBMIT & SHARE", variant="primary")
+            status_msg = gr.Markdown()
+            submit_btn.click(archive_submission, None, status_msg)
 
 demo.launch(allowed_paths=["."])