Update app.py

app.py

@@ -6,6 +6,7 @@ from transformers import pipeline
 import tempfile
 import os
 
+# Load and clean Titanic dataset
 def load_and_clean_data():
     url = "https://raw.githubusercontent.com/datasciencedojo/datasets/master/titanic.csv"
     df = pd.read_csv(url)
@@ -14,6 +15,7 @@ def load_and_clean_data():
     df.drop(columns=['Cabin', 'Ticket', 'Name'], inplace=True)
     return df
 
+# Generate EDA plots
 def generate_plot_images(df):
     temp_files = []
     for plot_func in [plot_survival_count, plot_survival_by_gender]:
@@ -31,6 +33,7 @@ def plot_survival_count(df, ax):
 def plot_survival_by_gender(df, ax):
     sns.countplot(x='Sex', hue='Survived', data=df, ax=ax)
 
+# Summarizer pipeline
 summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
 
 def summarize_eda(df):
@@ -45,15 +48,24 @@ def summarize_eda(df):
     except Exception as e:
         return f"Summarization failed: {str(e)}"
 
+# Markdown fallback logic
+def safe_markdown(df):
+    try:
+        return df.head().to_markdown()
+    except Exception:
+        return df.head().to_string()
+
+# Main app logic
 def run_titanic_eda():
     try:
         df = load_and_clean_data()
         plots = generate_plot_images(df)
         summary = summarize_eda(df)
-        return df.head().to_markdown(), plots[0], plots[1], summary
+        return safe_markdown(df), plots[0], plots[1], summary
     except Exception as e:
         return "Data load failed", None, None, f"Error: {str(e)}"
 
+# Gradio interface
 demo = gr.Interface(
     fn=run_titanic_eda,
     inputs=[],