Update app.py

app.py

@@ -7,9 +7,6 @@ import torch
 import matplotlib.pyplot as plt
 import seaborn as sns
 
-# Optional: Uncomment the following lines if you plan to use a gated model in the future
-# from huggingface_hub import login
-
 # Define constants
 MODEL_NAME = "gpt2"  # Publicly accessible model suitable for CPU
 FIGURES_DIR = "./figures"
@@ -20,7 +17,7 @@ EXAMPLE_FILE = os.path.join(EXAMPLE_DIR, "titanic.csv")
 os.makedirs(FIGURES_DIR, exist_ok=True)
 os.makedirs(EXAMPLE_DIR, exist_ok=True)
 
-# Download the Titanic dataset if it doesn't exist
+# Download the Titanic dataset if it doesn't exist (only for local runs)
 if not os.path.isfile(EXAMPLE_FILE):
     print("Downloading the Titanic dataset for examples...")
     try:
@@ -191,7 +188,16 @@ def interact_with_agent(file_input, additional_notes):
     for image_path in visualization_paths:
         # Ensure the image path is valid before attempting to display
         if os.path.isfile(image_path):
-            messages.append({"role": "assistant", "content": f"![{os.path.basename(image_path)}]({image_path})"})
+            # To display images in Gradio Chatbot, use the 'gr.Image' component
+            # However, since we're using 'type="messages"', we need to embed images via Markdown
+            # Gradio supports Markdown image embedding
+            image_filename = os.path.basename(image_path)
+            # Convert image path to relative path if necessary
+            # For Spaces, images should be in the repository
+            messages.append({
+                "role": "assistant",
+                "content": f"![{image_filename}]({image_path})"
+            })
         else:
             messages.append({"role": "assistant", "content": f"⚠️ Unable to find image: {image_path}"})