download

app.py

@@ -1,5 +1,6 @@
+import io
 import gradio as gr
-from io import StringIO
+import tempfile
 
 from uniprot_data import create_dataframe, get_uniprot_data
 
@@ -12,15 +13,17 @@ def process_uniprot_id(uniprot_id):
         uniprot_id: The UniProt ID.
 
     Returns:
-        A Pandas DataFrame or an error message.
+        Tuple of (DataFrame, downloadable file) or error message
     """
     protein_sequence, annotations = get_uniprot_data(uniprot_id)
 
     if protein_sequence and annotations:
         df = create_dataframe(protein_sequence, annotations)
-        return df
-    else:
-        return "Could not retrieve or process data for the given Uniprot ID"
+        # Create Excel file
+        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.xlsx')
+        df.to_excel(temp_file.name, index=False)
+        return df, temp_file.name
+    return "Could not retrieve or process data for the given Uniprot ID", None
 
 
 # Gradio Interface
@@ -58,8 +61,13 @@ with gr.Blocks() as demo:
         )
 
         output_df = gr.Dataframe(interactive=False)
+        download_btn = gr.DownloadButton(label="Download Excel")
 
-        submit_btn.click(fn=process_uniprot_id, inputs=input_text, outputs=output_df)
+        outputs = submit_btn.click(
+            fn=process_uniprot_id, 
+            inputs=input_text, 
+            outputs=[output_df, download_btn]
+        )
 
 if __name__ == "__main__":
     demo.launch()