Update app.py

app.py

@@ -293,13 +293,11 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         download_button = gr.File(label="Download Processed File", visible=False)
 
         def on_submit(llm, file, relevance, diversity):
-            result, output_file = process_inputs(llm, file, relevance, diversity)
-            if output_file:
-                return gr.File.update(value=output_file, visible=True)
-            else:
-                return gr.File.update(visible=False)
+            message, output_file = process_inputs(llm, file, relevance, diversity)
+            return message, output_file
+
+        submit_button.click(fn=on_submit, inputs=[llm_dropdown, file_upload, relevance_slider, diversity_slider], outputs=[None, download_button], postprocess=update_download_button)
 
-        submit_button.click(fn=on_submit, inputs=[llm_dropdown, file_upload, relevance_slider, diversity_slider], outputs=download_button)
             
 # Launch the Gradio app
 demo.launch()