Update app.py

app.py

@@ -13,15 +13,13 @@ csv_path = "cleaned_dataset_10k.csv"
 pkl_path = "final_embeddings_10k.pkl"
 
 if not os.path.exists(csv_path) or not os.path.exists(pkl_path):
-    # If files are missing, the app will show an error on startup
-    raise FileNotFoundError(f"Error: Required files not found in root directory.")
+    raise FileNotFoundError(f"Error: Files not found. I see: {os.listdir('.')}")
 
 # Load Data
 df = pd.read_csv(csv_path)
-# Standardize column names (lowercase, no spaces)
 df.columns = [c.strip().lower().replace(' ', '_') for c in df.columns]
 
-# Helper to identify correct column names automatically
+# Helper to find column names
 def get_col(candidates, default):
     for c in candidates:
         if c in df.columns: return c
@@ -32,86 +30,87 @@ col_rating = get_col(['rating', 'rating_score', 'stars'], 'rating')
 col_review = get_col(['review', 'review_content', 'review_content_clean'], 'review')
 col_persona = get_col(['reviewer_persona', 'persona', 'type'], 'reviewer_persona')
 
-# Load Pre-computed Embeddings
+# Load Embeddings
 with open(pkl_path, 'rb') as f:
     embedding_data = pickle.load(f)
-    dataset_embeddings = embedding_data['embeddings'] if isinstance(embedding_data, dict) else embedding_data
+    if isinstance(embedding_data, dict) and 'embeddings' in embedding_data:
+        dataset_embeddings = embedding_data['embeddings']
+    else:
+        dataset_embeddings = embedding_data
 
-# Load Semantic Model
+# Load Model
 model = SentenceTransformer('sentence-transformers/all-mpnet-base-v2')
 
-# Calculate Persona Taste Profiles (Mean Vectors)
+# Calculate Personas
 persona_profiles = {}
 if col_persona in df.columns:
     for persona in df[col_persona].unique():
         if pd.isna(persona): continue
         indices = df[df[col_persona] == persona].index
-        # Ensure indices are within embedding bounds
         valid_indices = [i for i in indices if i < len(dataset_embeddings)]
         if valid_indices:
-            persona_profiles[persona] = np.mean(dataset_embeddings[valid_indices], axis=0)
+            persona_vectors = dataset_embeddings[valid_indices]
+            persona_profiles[persona] = np.mean(persona_vectors, axis=0)
 else:
-    # Fallback if no persona column exists
     persona_profiles['Default'] = np.mean(dataset_embeddings, axis=0)
 
 # ==========================================
-# 2. RECOMMENDATION LOGIC
+# 2. LOGIC ENGINE
 # ==========================================
 def run_ven_engine(budget, dietary, company, purpose, noise):
-    # Construct user context string for embedding
     user_context = f"Searching for a {budget} experience, {dietary} friendly. Group: {company}. Occasion: {purpose}. Atmosphere: {noise}."
     query_vec = model.encode([user_context])
     
-    # Semantic similarity check against persona profiles
     similarities = {p: cosine_similarity(query_vec, v.reshape(1, -1))[0][0] for p, v in persona_profiles.items()}
     closest_persona = max(similarities, key=similarities.get)
     
-    # Filter dataset for the matched persona
-    persona_df = df[df[col_persona] == closest_persona] if col_persona in df.columns else df
-    if persona_df.empty: persona_df = df
+    if col_persona in df.columns:
+        persona_df = df[df[col_persona] == closest_persona]
+        if persona_df.empty: persona_df = df
+    else:
+        persona_df = df
     
-    # Selection Strategy: Highest rated restaurant for that persona
     top_match = persona_df.sort_values(by=col_rating, ascending=False).iloc[0]
+    
     match_pct = int(similarities[closest_persona] * 100)
-    review_text = str(top_match[col_review])[:180] + "..."
+    review_text = str(top_match[col_review])[:160] + "..."
     
-    # Return HTML Card for display
+    # --- VISUAL FIX ---
     return f"""
-    <div style="background: white; border-radius: 15px; padding: 20px; color: #1e293b !important; border-left: 8px solid #f97316; box-shadow: 0 10px 15px rgba(0,0,0,0.1);">
-        <div style="display:flex; justify-content:space-between; align-items: flex-start;">
+    <div style="background: white; border: 1px solid #e2e8f0; border-radius: 20px; padding: 24px; color: #000000 !important;">
+        <div style="display:flex; justify-content:space-between;">
             <div>
-                <div style="font-size: 24px; font-weight: 800; color: #0f172a !important; margin-bottom: 2px;">{top_match[col_name]}</div>
-                <div style="font-size: 14px; color: #64748b !important; font-weight: 600;">AI Match: {closest_persona} profile</div>
+                <div style="font-size: 22px; font-weight: 800; color: #000000 !important;">{top_match[col_name]}</div>
+                <div style="font-size: 14px; color: #333333 !important; font-weight: 600;">Match for: {closest_persona}</div>
             </div>
-            <div style="text-align:right; background: #f1f5f9; padding: 10px; border-radius: 10px;">
-                <div style="font-size: 26px; font-weight: 900; color: #f97316 !important;">{top_match[col_rating]}</div>
-                <div style="font-size:10px; font-weight:800; color: #475569 !important; letter-spacing: 1px;">RATING</div>
+            <div style="text-align:right;">
+                <div style="font-size: 28px; font-weight: 900; color: #2563eb !important;">{top_match[col_rating]}</div>
+                <div style="font-size:12px; font-weight:bold; color: #000000 !important;">RATING</div>
             </div>
         </div>
-        <div style="margin: 15px 0; font-size: 15px; font-style: italic; color: #334155 !important; line-height: 1.5;">
-            "{review_text}"
-        </div>
-        <div style="display: flex; justify-content: space-between; align-items: center; margin-top: 10px;">
-             <span style="font-size: 12px; font-weight: 700; color: #f97316;">Match Confidence: {match_pct}%</span>
-             <span style="font-size: 11px; background: #0f172a; color: white; padding: 3px 8px; border-radius: 5px;">VEN Matchmaker</span>
-        </div>
+        <hr style="border:0; border-top:1px solid #cbd5e1; margin: 15px 0;">
+        
+        <p style="color: #000000 !important; line-height:1.6; font-size: 16px; font-weight: 500; margin-top: 10px;">
+            <i style="color: #000000 !important;">"{review_text}"</i>
+        </p>
+        
+        <div style="margin-top:15px; font-size:13px; font-weight:700; color:#2563eb !important;">Match Confidence: {match_pct}%</div>
     </div>
     """
 
 # ==========================================
-# 3. UI & CSS (OPTIMIZED FOR VISIBILITY)
+# 3. APP UI & CSS FIX
 # ==========================================
-# Specific CSS to fix visibility of labels and radio buttons in dark mode
 ven_css = """
-.gradio-container { background-color: #0f172a !important; }
-/* Force labels above inputs to be white and visible */
-label span { color: white !important; font-weight: 600 !important; font-size: 14px !important; }
-/* Force radio button choice text to be white */
-.gr-radio label span { color: white !important; font-size: 13px !important; }
-/* Style the primary orange button */
-.ven-button { background-color: #f97316 !important; color: white !important; border: none !important; font-weight: 800 !important; }
-/* Ensure headings are white */
-h1 { color: white !important; text-align: center; }
+body { background-color: #0f172a !important; font-family: sans-serif !important; }
+/* Global White Text for Dark Mode */
+h1, h2, h3, h4, h5, h6 { color: white !important; }
+p, span, div, label { color: white; }
+
+/* Override: Force Black Text inside the Results Card */
+.gradio-html div { color: #000000 !important; }
+.gradio-html p { color: #000000 !important; }
+.gradio-html i { color: #000000 !important; }
 """
 
 with gr.Blocks(css=ven_css, title="VEN Project") as demo:
@@ -119,17 +118,16 @@ with gr.Blocks(css=ven_css, title="VEN Project") as demo:
     
     with gr.Row():
         with gr.Column():
-            with gr.Group(): # Group keeps labels and inputs contained
-                in_budget = gr.Dropdown(["Budget-friendly", "Mid-range", "Premium"], label="1. Select Budget", value="Mid-range")
-                in_diet = gr.Dropdown(["Anything", "Vegetarian", "Vegan", "Meat-lover"], label="2. Dietary Preference", value="Anything")
-                in_company = gr.Radio(["Solo", "Date/Couple", "Friends", "Business"], label="3. Who are you with?", value="Date/Couple")
-                in_purpose = gr.Dropdown(["Casual dinner", "Special occasion", "Quick bite", "Professional meeting"], label="4. Occasion", value="Casual dinner")
-                in_noise = gr.Radio(["Quiet/Intimate", "Moderate/Social", "Lively/Music"], label="5. Environment vibe", value="Moderate/Social")
-                btn = gr.Button("Find My Table", variant="primary", elem_classes="ven-button")
-        
+            in_budget = gr.Dropdown(["Budget-friendly", "Mid-range", "Premium"], label="Budget", value="Mid-range")
+            in_diet = gr.Dropdown(["Anything", "Vegetarian", "Vegan", "Meat-lover"], label="Diet", value="Anything")
+            in_company = gr.Radio(["Solo", "Date/Couple", "Friends", "Business"], label="With who?", value="Date/Couple")
+            in_purpose = gr.Dropdown(["Casual dinner", "Special occasion", "Quick bite", "Professional meeting"], label="Occasion", value="Casual dinner")
+            in_noise = gr.Radio(["Quiet/Intimate", "Moderate/Social", "Lively/Music"], label="Environment", value="Moderate/Social")
+            btn = gr.Button("Find My Table", variant="primary")
+
         with gr.Column():
-            output_ui = gr.HTML("<div style='text-align:center; padding:50px; color:#94a3b8;'>Your personalized recommendation will appear here...</div>")
-    
+            output_ui = gr.HTML("<h4>Recommendation will appear here...</h4>")
+
     gr.Markdown("### 🚀 Quick Starters (One-Click)")
     gr.Examples(
         examples=[
@@ -140,9 +138,9 @@ with gr.Blocks(css=ven_css, title="VEN Project") as demo:
         inputs=[in_budget, in_diet, in_company, in_purpose, in_noise],
         outputs=output_ui,
         fn=run_ven_engine,
-        cache_examples=False, # Disable cache if you want real-time testing
+        cache_examples=True,
     )
-    
+
     btn.click(run_ven_engine, inputs=[in_budget, in_diet, in_company, in_purpose, in_noise], outputs=output_ui)
 
 if __name__ == "__main__":