Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 from PIL import Image
+import os
 
 # Set page config to make the background dark blue
 st.set_page_config(page_title="My App", layout="centered")
@@ -16,19 +17,45 @@ st.markdown(
     unsafe_allow_html=True
 )
 
-# Define function to load and display images
+# Base path for images and HTML files in the Hugging Face Spaces repository
+base_path = "Karlsen/Tweet_data/"
+
+# Define function to load and display images with error handling
 def display_image(image_path, caption):
-    image = Image.open(image_path)
-    st.image(image, caption=caption, use_column_width=True)
+    try:
+        image = Image.open(image_path)
+        st.image(image, caption=caption, use_column_width=True)
+    except FileNotFoundError:
+        st.error(f"File not found: {image_path}")
+    except Exception as e:
+        st.error(f"An error occurred while loading {image_path}: {e}")
+
+# List of image files with their captions
+image_files = [
+    ("head10.jpeg", "Head 10"),
+    ("accuracy.jpeg", "Accuracy"),
+    ("heirarchy_cluster.jpeg", "Heirarchy Cluster"),
+    ("sentiment.jpeg", "Sentiment"),
+    ("similarity_matrix.jpeg", "Similarity Matrix"),
+    ("topic_groups.jpeg", "Topic Groups")
+]
 
-# Display images
-display_image("head10.jpeg", "Head 10")
-display_image("accuracy.jpeg", "Accuracy")
-display_image("heirarchy_cluster.jpeg", "Heirarchy Cluster")
-display_image("sentiment.jpeg", "Sentiment")
-display_image("similarity_matrix.jpeg", "Similarity Matrix")
-display_image("topic_groups.jpeg", "Topic Groups")
+# Check if the images are in the correct directory
+missing_files = [file for file, _ in image_files if not os.path.exists(base_path + file)]
 
-# Display interactive HTML file
-st.components.v1.html(open("bertopic_visualization.html", "r").read(), height=800)
+if missing_files:
+    st.error(f"Missing files: {', '.join(missing_files)}")
+else:
+    # Display images
+    for file, caption in image_files:
+        display_image(base_path + file, caption)
 
+# Display interactive HTML file with error handling
+html_file = base_path + "bertopic_visualization.html"
+if os.path.exists(html_file):
+    try:
+        st.components.v1.html(open(html_file, "r").read(), height=800)
+    except Exception as e:
+        st.error(f"An error occurred while loading {html_file}: {e}")
+else:
+    st.error(f"File not found: {html_file}")