updated gradio to call the functions directly

app.py

@@ -5,6 +5,7 @@ import pandas as pd
 import matplotlib.pyplot as plt
 import time
 import io
+from PIL import Image
 
 def analyze_file(file, progress=gr.Progress(track_tqdm=True)):
     start_time = time.time()
@@ -28,10 +29,11 @@ def analyze_file(file, progress=gr.Progress(track_tqdm=True)):
     plt.savefig(buf, format='png')
     buf.seek(0)
     plt.close(fig)
+    img = Image.open(buf)
     progress(1, desc="Done!")
     elapsed = time.time() - start_time
     preview = df.head().to_csv(index=False)
-    return preview, csv_path, buf, f"Analysis completed in {elapsed:.2f} seconds."
+    return preview, csv_path, img, f"Analysis completed in {elapsed:.2f} seconds."
 
 iface = gr.Interface(
     fn=analyze_file,