Update app.py

app.py

@@ -5,12 +5,22 @@ from transformers import pipeline
 # Load the WikiArt dataset in streaming mode
 dataset = load_dataset("huggan/wikiart", streaming=True)
 
+# Function to safely get a field value or return a default message
+def get_field(record, field, default="Unknown"):
+    return record[field] if field in record and record[field] is not None else default
+
 # Function to display artwork details
 def display_artwork(index):
     try:
         for i, record in enumerate(dataset["train"]):  # Stream through the dataset
             if i == index:
-                return record["image"], f"Title: {record['title']}\nArtist: {record['artist']}\nStyle: {record['style']}\nGenre: {record['genre']}"
+                return (
+                    get_field(record, "image"), 
+                    f"Title: {get_field(record, 'title')}\n"
+                    f"Artist: {get_field(record, 'artist')}\n"
+                    f"Style: {get_field(record, 'style')}\n"
+                    f"Genre: {get_field(record, 'genre')}"
+                )
         return None, "Error: Index out of range or invalid."
     except Exception as e:
         return None, f"Error: {str(e)}"
@@ -20,9 +30,9 @@ def filter_artworks(artist=None, genre=None, style=None):
     results = []
     try:
         for record in dataset["train"]:
-            if (artist is None or record["artist"] == artist) and \
-               (genre is None or record["genre"] == genre) and \
-               (style is None or record["style"] == style):
+            if (artist is None or get_field(record, "artist") == artist) and \
+               (genre is None or get_field(record, "genre") == genre) and \
+               (style is None or get_field(record, "style") == style):
                 results.append(record)
     except Exception as e:
         return []
@@ -33,7 +43,12 @@ def display_filtered_artworks(artist, genre, style):
     filtered_results = filter_artworks(artist, genre, style)
     if len(filtered_results) == 0:
         return None, "No artworks found with the specified filters."
-    return [(r["image"], f"Title: {r['title']}\nArtist: {r['artist']}\nStyle: {r['style']}\nGenre: {r['genre']}") for r in filtered_results]
+    return [(get_field(r, "image"), 
+             f"Title: {get_field(r, 'title')}\n"
+             f"Artist: {get_field(r, 'artist')}\n"
+             f"Style: {get_field(r, 'style')}\n"
+             f"Genre: {get_field(r, 'genre')}")
+            for r in filtered_results]
 
 # Chatbot functionality for museum guide persona using a publicly available Hugging Face model
 chatbot = pipeline("text-generation", model="gpt2")  # Replace with a valid Hugging Face model