Update app.py

app.py

@@ -1050,8 +1050,10 @@ def submit_predict(predict_filepath, task, preset, target_family, opts, job_info
 
             max_sim.cache_clear()
 
-        prediction_df.drop(['N'], axis=1).to_csv(predictions_file, index=False, na_rep='')
-        status = "COMPLETED"
+        prediction_df.drop(
+            [col for col in ['N', 'FP'] if col in prediction_df.columns], axis=1
+        ).to_csv(predictions_file, index=False, na_rep='')
+        status = 'COMPLETED'
 
         return {run_state: False}
 
@@ -2750,13 +2752,16 @@ QALAHAYFAQYHDPDDEPVADPYDQSFESRDLLIDEWKSLTYDEVISFVPPPLDQEEMES
         else:
             return {report_task: gr.Dropdown(visible=False)}
 
-
     report_df_change = file_for_report.change(
         fn=update_df, inputs=file_for_report, outputs=[html_report, raw_df, report_df, analyze_btn, report_task],
         concurrency_limit=100,
     ).success(
-        fn=lambda: [gr.Button(interactive=True)] * 2,
-        outputs=[csv_generate, html_generate],
+        fn=lambda: [gr.Button(interactive=True)] * 3 +
+                   [gr.File(visible=False, value=None)] * 2 +
+                   [gr.Dropdown(visible=False, value=None), ''],
+        outputs=[
+            csv_generate, html_generate, analyze_btn, csv_download_file, html_download_file, report_task, html_report
+        ],
     )
 
     file_for_report.upload(