Update app.py

app.py

@@ -108,8 +108,8 @@ def update(sequence=DEFAULT_SEQ):
     tmp = tempfile.NamedTemporaryFile()
     with open(tmp.name, "w") as f:
         f.write(pdb_string)
-
-    return molecule(tmp.name)
+    print("File name", tmp.name)
+    return molecule(tmp.name), tmp.name
 
 
 def suggest(option):
@@ -152,8 +152,8 @@ with demo:
         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)
     mol = gr.HTML(update)    
-    #download = gr.File(label="Download file")
-    btn.click(fn=update, inputs=inp, outputs=mol)
+    download = gr.File(label="Download file")
+    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)