|
|
import tempfile |
|
|
import pandas as pd |
|
|
import gradio as gr |
|
|
import json |
|
|
import plotly.io as pio |
|
|
|
|
|
|
|
|
class DownloadHelper: |
|
|
@staticmethod |
|
|
def df2csv(df: pd.DataFrame, prefix: str): |
|
|
with tempfile.NamedTemporaryFile(mode="w", delete=False, prefix=f"{prefix}_", |
|
|
suffix=".csv", encoding="utf-8") as tmpfile: |
|
|
df.to_csv(tmpfile.name, index=False) |
|
|
return tmpfile.name |
|
|
|
|
|
@staticmethod |
|
|
def fig2png(fig: gr.components.plot.PlotData, prefix: str): |
|
|
with tempfile.NamedTemporaryFile(mode="wb", delete=False, prefix=f"{prefix}_", |
|
|
suffix=".png") as tmpfile: |
|
|
pio.write_image(json.loads(fig.plot), tmpfile.name, format="png") |
|
|
return tmpfile.name |
|
|
|
