Update app.py

app.py

@@ -66,6 +66,8 @@ def molecule(input_pdb):
     allow-top-navigation-by-user-activation allow-downloads" allowfullscreen="" 
     allowpaymentrequest="" frameborder="0" srcdoc='{x}'></iframe>"""
 
+import tempfile
+
 
 def update(sequence=DEFAULT_SEQ):
     headers = {
@@ -76,10 +78,10 @@ def update(sequence=DEFAULT_SEQ):
     name = sequence[:3] + sequence[-3:] 
     pdb_filename = "test.pdb"
     pdb_string = response.content.decode('utf-8')
-    with open(pdb_filename, "w") as out:
-      out.write(pdb_string)
+    with tempfile.NamedTemporaryFile() as tmp:
+        tmp.write(pdb_string)
 
-    return molecule(pdb_filename)
+    return molecule(tmp.name), tmp.name
 
 
 def suggest(option):
@@ -120,9 +122,13 @@ with demo:
     name = gr.Dropdown(label="Choose a Sample Protein", value="Plastic degradation protein", choices=["Antifreeze protein", "Plastic degradation protein",  "AI Generated protein", "7-bladed propeller fold", "custom"])
     with gr.Row():
         inp = gr.Textbox(label="Protein sequence", lines=3, value=DEFAULT_SEQ, placeholder="Write your protein sequence here...")
-        btn = gr.Button("🔬 Predict Structure ").style(full_width=False)
+        with gr.Column():
+            with gr.Row():
+                btn = gr.Button("🔬 Predict Structure ").style(full_width=False)
+            with gr.Row():
+                download = gr.File(label="Download file")
     mol = gr.HTML(update)    
-    btn.click(fn=update, inputs=inp, outputs=mol)
+    btn.click(fn=update, inputs=inp, outputs=[mol, download])
     
     name.change(fn=suggest, inputs=name, outputs=inp)
     name.change(fn=lambda :"", inputs=None, outputs=mol)