Update app.py

app.py

@@ -12,6 +12,14 @@ DATASET_ID = "motimmom/cocktails_clean_nobrand"   # <- change if needed
 EMBED_MODEL = "sentence-transformers/all-MiniLM-L6-v2"
 FLAVOR_BOOST = 0.20
 
+# Backgrounds (override default with env var BG_URL)
+BACKGROUND_CHOICES = {
+    "Tiki":    "https://images.unsplash.com/photo-1514362545857-3bc16c4c76cf?q=80&w=1920&auto=format&fit=crop",
+    "Martini": "https://images.unsplash.com/photo-1536935339748-021c8d927f21?q=80&w=1920&auto=format&fit=crop",
+    "Negroni": "https://images.unsplash.com/photo-1551024709-8f23befc6cf7?q=80&w=1920&auto=format&fit=crop",
+}
+DEFAULT_BG_URL = os.getenv("BG_URL") or BACKGROUND_CHOICES["Negroni"]
+
 # If dataset is private, add Space secret HF_TOKEN (read scope)
 HF_READ_TOKEN = os.getenv("HF_TOKEN") or os.getenv("HUGGING_FACE_HUB_TOKEN")
 load_kwargs = {}
@@ -243,16 +251,14 @@ def _split_measure_name_line(line: str):
 
 def _format_ingredients_markdown(lines):
     """
-    Bullet points with 'Ingredient (amount)'. Also removes any [ or ].
+    Bullet points as 'Ingredient (amount)'. Strips [ and ] if present.
     """
     if not lines:
         return "—"
     formatted = []
     for ln in lines:
-        # Strip any square brackets the dataset might include
         ln = ln.replace("[", "").replace("]", "")
         meas, name = _split_measure_name_line(ln)
-        # Flip to Ingredient (amount)
         if name and meas:
             formatted.append(f"- {name} ({meas})")
         elif name:
@@ -306,28 +312,62 @@ def recommend(base_alcohol_text, flavor, top_k=3):
     return "\n\n---\n\n".join(blocks)
 
 # ========================
-# UI
+# Background + UI
 # ========================
-with gr.Blocks() as demo:
-    gr.Markdown("# 🍹 AI Bartender — Type a Base + Flavor (Clear Ingredients)")
-
-    with gr.Row():
-        base_text = gr.Textbox(value="gin", label="Base alcohol (type any spirit, e.g., 'gin', 'white rum', 'bourbon')")
-        flavor = gr.Dropdown(choices=FLAVOR_OPTIONS, value="citrus", label="Flavor")
-        topk = gr.Slider(1, 10, value=3, step=1, label="Number of recommendations")
-
-    with gr.Row():
-        ex1 = gr.Button("Example: Gin + Citrus")
-        ex2 = gr.Button("Example: Rum + Fruity")
-        ex3 = gr.Button("Example: Mezcal + Smoky")
-
-    out = gr.Markdown()
-    gr.Button("Recommend").click(recommend, [base_text, flavor, topk], out)
-
-    # Quick-fill examples
-    ex1.click(lambda: ("gin", "citrus", 3), outputs=[base_text, flavor, topk])
-    ex2.click(lambda: ("white rum", "fruity", 3), outputs=[base_text, flavor, topk])
-    ex3.click(lambda: ("mezcal", "smoky", 3), outputs=[base_text, flavor, topk])
+def render_bg_div(url: str) -> str:
+    return f"<div id='app-bg' style=\"position:fixed;inset:0;z-index:-1;background-image:url('{url}');background-size:cover;background-position:center;filter:brightness(0.45);\"></div>"
+
+CUSTOM_CSS = """
+.gradio-container { background: transparent !important; }
+.glass-card {
+  background: rgba(255, 255, 255, 0.08);
+  backdrop-filter: blur(6px);
+  -webkit-backdrop-filter: blur(6px);
+  border-radius: 14px;
+  padding: 18px;
+  border: 1px solid rgba(255, 255, 255, 0.12);
+}
+"""
+
+def choose_background(choice, custom_url):
+    if choice in BACKGROUND_CHOICES:
+        url = BACKGROUND_CHOICES[choice]
+    else:
+        url = custom_url.strip() or DEFAULT_BG_URL
+    return render_bg_div(url)
+
+with gr.Blocks(css=CUSTOM_CSS) as demo:
+    bg_html = gr.HTML(render_bg_div(DEFAULT_BG_URL))
+
+    with gr.Column(elem_classes=["glass-card"]):
+        gr.Markdown("# 🍹 AI Bartender — Type a Base + Flavor (Clear Ingredients)")
+
+        with gr.Row():
+            base_text = gr.Textbox(value="gin", label="Base alcohol (type any spirit, e.g., 'gin', 'white rum', 'bourbon')")
+            flavor = gr.Dropdown(choices=FLAVOR_OPTIONS, value="citrus", label="Flavor")
+            topk = gr.Slider(1, 10, value=3, step=1, label="Number of recommendations")
+
+        with gr.Row():
+            ex1 = gr.Button("Example: Gin + Citrus")
+            ex2 = gr.Button("Example: Rum + Fruity")
+            ex3 = gr.Button("Example: Mezcal + Smoky")
+
+        out = gr.Markdown()
+        gr.Button("Recommend").click(recommend, [base_text, flavor, topk], out)
+
+        gr.Markdown("### 🎨 Background")
+        with gr.Row():
+            bg_choice = gr.Radio(choices=list(BACKGROUND_CHOICES.keys()) + ["Custom URL"], value="Negroni", label="Theme")
+            bg_custom = gr.Textbox(label="Custom background image URL", placeholder="https://... (used only if 'Custom URL' is selected)")
+
+        # Update background on change
+        bg_choice.change(choose_background, [bg_choice, bg_custom], bg_html)
+        bg_custom.change(choose_background, [bg_choice, bg_custom], bg_html)
+
+        # Quick-fill examples
+        ex1.click(lambda: ("gin", "citrus", 3), outputs=[base_text, flavor, topk])
+        ex2.click(lambda: ("white rum", "fruity", 3), outputs=[base_text, flavor, topk])
+        ex3.click(lambda: ("mezcal", "smoky", 3), outputs=[base_text, flavor, topk])
 
 if __name__ == "__main__":
     demo.launch()