Update app.py

app.py

@@ -9,7 +9,7 @@ import random
 from sentence_transformers import SentenceTransformer
 from sklearn.metrics.pairwise import cosine_similarity
 
-# --- 1. הכנת תמונות (סנכרון תיקייה פנימית) ---
+# --- 1. הכנת תמונות ---
 IMAGE_DIR = "extracted_images"
 if os.path.exists('images.zip'):
     with zipfile.ZipFile('images.zip', 'r') as zip_ref:
@@ -40,7 +40,7 @@ def load_data():
 main_df, dish_embeddings, user_embeddings = load_data()
 NAMES = ["James Miller", "Sarah Johnson", "Michael Brown", "Emily Davis", "Robert Wilson"]
 
-# --- 3. מנוע החיפוש (Discovery) ---
+# --- 3. מנוע החיפוש ---
 def run_discovery(query, origin, hobbies, style):
     q_vec = model.encode([str(query)])
     u_dna = f"Origin: {origin}, Hobbies: {hobbies}, Style: {style}"
@@ -69,22 +69,20 @@ def run_discovery(query, origin, hobbies, style):
         img_url = find_dish_image(original_idx)
         
         html_output += f"""
-        <div style="border: 1px solid #C4A484; border-radius: 4px; padding: 25px; margin-bottom: 30px; background: #FFF9F5; border-left: 10px solid #3E2723; display: flex; gap: 25px; color: #3E2723; box-shadow: 5px 5px 15px rgba(0,0,0,0.05);">
+        <div style="border: 1px solid #C4A484; padding: 25px; margin-bottom: 30px; background: #FFF9F5; border-left: 10px solid #3E2723; display: flex; gap: 25px; color: #3E2723;">
             <img src="{img_url}" style="width: 250px; height: 250px; object-fit: cover; border: 1px solid #D2B48C; padding: 4px; background: white;">
             <div style="flex: 1;">
                 <div style="display: flex; justify-content: space-between; align-items: center;">
-                    <h2 style="margin: 0; color: #3E2723; font-family: 'Playfair Display', serif; font-size: 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>
+                    <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;">{pct} MATCH</span>
                 </div>
                 <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.1em; line-height: 1.4;">"{row['taste_review']}"</p>
-                
+                <p style="font-family: 'Playfair Display', serif; font-style: italic; color: #4E342E; margin: 15px 0; font-size: 1.1em;">"{row['taste_review']}"</p>
                 <div style="display: flex; gap: 10px; margin-bottom: 15px;">
-                    <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; border: 1px solid #D2B48C;">💰 PRICE: {row.get('price_range', '$$')}</span>
+                    <span style="background: #EEDDCC; color: #3E2723; padding: 4px 10px; border: 1px solid #D2B48C;">👥 BEST FOR: {row.get('best_for', 'Friends')}</span>
                 </div>
-
-                <div style="background: #F5F1EE; padding: 12px; border-radius: 4px; font-size: 0.9em; color: #5D4037;">
+                <div style="background: #F5F1EE; padding: 12px; font-size: 0.9em; border-radius: 4px;">
                     <b>Twin:</b> {u_name} from {row['user_origin']} | <b>Rating:</b> {row['rating']}/5.0
                 </div>
             </div>
@@ -92,44 +90,56 @@ def run_discovery(query, origin, hobbies, style):
         """
     return html_output
 
-def archive_msg(): return "### Thank you! \nYour review has been successfully shared with the BiteWise archive."
-
-# --- 4. הממשק המעוצב (3 מסכים) ---
+# --- 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; }
-button.primary { background: #3E2723 !important; color: white !important; border-radius: 0px !important; font-family: 'Courier New', monospace; text-transform: uppercase; letter-spacing: 2px; }
-input, .dropdown { border-radius: 0px !important; border: 1px solid #D2B48C !important; background: white !important; }
+/* הסתרת הטאבים למראה של אפליקציה אחת חלקה */
+.tabs > .tab-nav { display: none !important; }
+.gradio-container { background-color: #FDFCF8 !important; color: #3E2723; }
+button.primary { background: #3E2723 !important; border-radius: 0px !important; color: white !important; font-family: 'Courier New', monospace; text-transform: uppercase; }
+input, .dropdown { border-radius: 0px !important; border: 1px solid #D2B48C !important; }
 """
 
 with gr.Blocks(css=custom_css) as demo:
     gr.HTML("<h1 style='text-align: center; color: #3E2723; font-family: \"Playfair Display\", serif; font-size: 4em; font-style: italic; border-bottom: 2px double #D2B48C; margin-bottom: 40px;'>BiteWise</h1>")
     
     with gr.Tabs() as tabs:
-        with gr.Tab("01. THE PROFILE", id=0):
+        # מסך DNA
+        with gr.TabItem("DNA", id=0):
             with gr.Row():
-                u_n = gr.Textbox(label="IDENTIFICATION (NAME)")
+                u_n = gr.Textbox(label="IDENTIFICATION")
                 u_o = gr.Dropdown(list(main_df['user_origin'].unique()), label="ORIGIN", value="Tel Aviv")
             with gr.Row():
                 u_h = gr.Dropdown(list(main_df['user_hobbies'].unique()), label="INTERESTS")
-                u_s = gr.Dropdown(list(main_df['user_fashion_style'].unique()) + ["Vintage/Retro"], label="AESTHETIC STYLE", value="Vintage/Retro")
+                u_s = gr.Dropdown(list(main_df['user_fashion_style'].unique()) + ["Vintage/Retro"], label="STYLE", value="Vintage/Retro")
             btn1 = gr.Button("SYNC PERSONALITY", variant="primary")
-            btn1.click(lambda: gr.update(selected=1), None, tabs)
+            btn1.click(lambda: gr.Tabs(selected=1), None, tabs)
 
-        with gr.Tab("02. DISCOVERY", id=1):
+        # מסך DISCOVERY
+        with gr.TabItem("DISCOVERY", id=1):
             q_in = gr.Textbox(label="CRAVING", placeholder="What are you in the mood for?")
-            btn2 = gr.Button("COMMENCE SEARCH", variant="primary")
+            btn2 = gr.Button("SEARCH ARCHIVE", variant="primary")
             out_html = gr.HTML()
+            with gr.Row():
+                btn_back = gr.Button("BACK TO DNA")
+                btn_to_archive = gr.Button("ADD NEW DISCOVERY")
+            
             btn2.click(run_discovery, [q_in, u_o, u_h, u_s], out_html)
+            btn_back.click(lambda: gr.Tabs(selected=0), None, tabs)
+            btn_to_archive.click(lambda: gr.Tabs(selected=2), None, tabs)
 
-        with gr.Tab("03. ARCHIVE", id=2):
+        # מסך ARCHIVE
+        with gr.TabItem("ARCHIVE", id=2):
             gr.Markdown("### ARCHIVE A NEW CULINARY DISCOVERY")
             with gr.Row():
                 gr.Textbox(label="DISH")
                 gr.Textbox(label="ESTABLISHMENT")
             gr.Textbox(label="REVIEW", lines=3)
-            s_btn = gr.Button("SUBMIT & SHARE", variant="primary")
-            s_msg = gr.Markdown()
-            s_btn.click(archive_msg, None, s_msg)
+            s_btn = gr.Button("SUBMIT TO ARCHIVE", variant="primary")
+            status_msg = gr.Markdown()
+            btn_back_disc = gr.Button("BACK TO DISCOVERY")
+            
+            s_btn.click(lambda: "### Thank you! \nYour review has been successfully shared.", None, status_msg)
+            btn_back_disc.click(lambda: gr.Tabs(selected=1), None, tabs)
 
 demo.launch(allowed_paths=["."])