Update app.py

app.py

@@ -287,11 +287,10 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
         def on_submit(llm, file, relevance, diversity):
             message, output_file = process_inputs(llm, file, relevance, diversity)
-            if output_file:
-                download_button.update(value=output_file, visible=True)
-            return message
+            return message, update_download_button(output_file)
 
-        submit_button.click(fn=on_submit, inputs=[llm_dropdown, file_upload, relevance_slider, diversity_slider], outputs=[output_message])
+
+        submit_button.click(fn=on_submit, inputs=[llm_dropdown, file_upload, relevance_slider, diversity_slider], outputs=[output_message, download_button])
    
 # Launch the Gradio app
 demo.launch()