Update app.py

app.py

@@ -3,22 +3,18 @@ import pandas as pd
 from transformers import pipeline
 from PIL import Image
 import requests
-import os
 
-# ---- SETUP SECTION ----
+# ---- SETUP ----
+# Load dataset (make sure 'gldv2_info.csv' is in the project root)
+df = pd.read_csv("gldv2_info.csv")
 
-# Load your landmarks dataset
-# Place "gldv2_info.csv" in the root of your Space
-df = pd.read_csv("gldv2_info.csv")  # columns: gldv2_id, url, landmark_id, name
-
-# Text/story generator: use a free/fast mythology model from HF
+# Load a free mythology/historical text generator from Hugging Face
 try:
     story_pipe = pipeline("text-generation", model="Samurai719214/gptneo-mythology-storyteller")
 except Exception:
-    # Fallback model if main one fails
     story_pipe = pipeline("text-generation", model="mahing/historical-narrative-generator")
 
-# Image generator: Stable Diffusion (for cases where no matching image is found)
+# Optional: set up text-to-image only if 'diffusers' is available and desired
 try:
     from diffusers import StableDiffusionPipeline
     import torch
@@ -30,12 +26,12 @@ try:
 except Exception:
     sd_pipe = None
 
-# ---- APP LOGIC ----
-
+# ---- APP FUNCTION ----
 def landmark_explorer(site_name):
-    # 1. Find matching images
-    matches = df[df['name'].str.strip().str.lower() == site_name.strip().lower()]
+    site_name_clean = site_name.strip().lower()
+    matches = df[df['name'].str.strip().str.lower() == site_name_clean]
     images = []
+    # Gather all valid images
     for url in matches['url']:
         try:
             img = Image.open(requests.get(url, stream=True, timeout=6).raw).convert("RGB")
@@ -43,7 +39,7 @@ def landmark_explorer(site_name):
         except Exception:
             continue
 
-    # 2. Generate story
+    # Generate historical/myth story
     story_prompt = (
         f"Tell a mythological and historical story about the site: {site_name} in Egypt. "
         "Include both legend and real archaeological facts where possible."
@@ -54,11 +50,10 @@ def landmark_explorer(site_name):
     except Exception:
         story = "Unable to generate story for this landmark at the moment."
 
-    # 3. If no images, generate a relevant one (optional)
+    # If no images, generate one with Stable Diffusion (optional)
     if not images and sd_pipe is not None:
         try:
-            sd_out = sd_pipe(f"{site_name} in ancient Egypt, detailed, realistic, landscape")[0]
-            images = [sd_out]
+            images = [sd_pipe(f"{site_name} in ancient Egypt, detailed, realistic, landscape").images[0]]
         except Exception:
             pass
 
@@ -66,8 +61,7 @@ def landmark_explorer(site_name):
         return story, None
     return story, images
 
-# ---- GRADIO INTERFACE ----
-
+# ---- GRADIO UI ----
 with gr.Blocks() as demo:
     gr.Markdown(
         """
@@ -77,7 +71,7 @@ with gr.Blocks() as demo:
     )
     name_input = gr.Textbox(label="Enter Landmark Name (e.g., Great_Pyramid_of_Giza)")
     story_output = gr.Textbox(label="Generated Story")
-    gallery_output = gr.Gallery(label="Images").style(grid=3)
+    gallery_output = gr.Gallery(label="Images", columns=3)
     run_btn = gr.Button("Explore")
 
     run_btn.click(
@@ -87,10 +81,8 @@ with gr.Blocks() as demo:
     )
 
     gr.Markdown(
-        "Sample names: Great_Pyramid_of_Giza, Karnak, Temple_of_Edfu, Bab_al-Nasr_(Cairo), etc."
+        "Sample landmark names: Great_Pyramid_of_Giza, Karnak, Temple_of_Edfu, Bab_al-Nasr_(Cairo), etc."
     )
 
-# ---- LAUNCH APP ----
-
 if __name__ == "__main__":
     demo.launch()