Update app.py

app.py

@@ -149,21 +149,20 @@ def inference_dataset5(input_file, input_checks):
         return gr.update(value="This option was not selected.")
 """
 
-def file(input_file):
+def file(input_file, input_checks):
     output = "output.txt"
     f = open(output, 'w')
     f.write("The predictions come here.")
     f.close()
-    return gr.update(value=output, visible=True)
+    if "emotion frequencies" in input_checks:
+        return gr.update(value=output, visible=True), gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)  # next_button_freq becomes available
+    elif "emotion distribution over time" in input checks:
+        return gr.update(value=output, visible=True), gr.update(visible=False), gr.update(visible=True), gr.update(visible=False), gr.update(visible=False)  # next_button_dist becomes available
+    elif "peaks" in input checks:
+        return gr.update(value=output, visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=True), gr.update(visible=False)  # next_button_peaks becomes available
+    elif "topics" in input checks:
+        return gr.update(value=output, visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=True)  # next_button_peaks becomes available
 
-def file_next(input_file):
-    output = "output.txt"
-    f = open(output, 'w')
-    f.write("The predictions come here.")
-    f.close()
-    return gr.update(value=output, visible=True), gr.update(visible=True) # next_button becomes available
-                
-        
 
 
 with gr.Blocks() as demo:
@@ -209,17 +208,7 @@ with gr.Blocks() as demo:
                 output_topics = gr.Textbox(label="Topics", visible=False)
         
         #send_btn.click(fn=inference_dataset1, inputs=input_file, outputs=[output_file,next_button_freq]) # next_button_freq becomes visible
-        
-        if "emotion frequencies" in input_checks:
-            send_btn.click(fn=file_next, inputs=input_file, outputs=[output_file,next_button_freq]) # next_button_freq becomes visible
-        elif "emotion distribution over time" in input_checks:
-            send_btn.click(fn=file_next, inputs=input_file, outputs=[output_file,next_button_dist]) # next_button_dist becomes visible
-        elif "peaks" in input_checks:
-            send_btn.click(fn=file_next, inputs=input_file, outputs=[output_file,next_button_peaks]) # next_button_peaks becomes visible
-        elif "topics" in input_checks:
-            send_btn.click(fn=file_next, inputs=input_file, outputs=[output_file,next_button_topics]) # next_button_topics becomes visible
-        else:
-            send_btn.click(fn=file, inputs=input_file, outputs=output_file) # no next_button becomes available
+        send_btn.click(fn=file, inputs=input_file, outputs=[output_file,next_button_freq,next_button_dist,next_button_peaks,next_button_topics])
 
         """"
         if "emotion frequencies" in input_checks: