v.1.44

app.py

@@ -643,9 +643,10 @@ def create_output_file(df, uploaded_file):
 
 def create_interface():
     control = ProcessControl()
-    
+    file_data = gr.State(value=None)
+
     with gr.Blocks(theme=gr.themes.Soft()) as app:
-        gr.Markdown("# AI-анализ мониторинга новостей v.1.43")
+        gr.Markdown("# AI-анализ мониторинга новостей v.1.44")
         
         with gr.Row():
             file_input = gr.File(
@@ -701,6 +702,11 @@ def create_interface():
                 visible=False
             )
         
+            file_output = gr.File(
+                label="Скачать результаты",
+                visible=False
+            )
+
         def stop_processing():
             control.request_stop()
             return "Остановка обработки..."
@@ -791,16 +797,18 @@ def create_interface():
                 if processed_rows:
                     result_df = pd.DataFrame(processed_rows)
                     output_bytes_io = create_output_file(result_df, file_obj)
+                    fig_sentiment, fig_events = create_visualizations(result_df)
+
                     if output_bytes_io:
-                        fig_sentiment, fig_events = create_visualizations(result_df)
-                        yield (
+                        output_bytes = output_bytes_io.getvalue()
+                        return (
                             result_df,
                             fig_sentiment,
                             fig_events,
-                            True,
-                            #(f"results_{len(processed_rows)}_rows.xlsx", output_bytes_io.getvalue()),  # Return tuple of (name, bytes)
+                            gr.update(visible=True),
                             f"Обработано {len(processed_rows)}/{total} строк",
-                            dedup_message
+                            dedup_message, 
+                            (output_bytes, "результаты_анализа.xlsx")
                         )
                 
                 return (
@@ -809,7 +817,8 @@ def create_interface():
                     None,  # events plot
                     gr.update(visible=False),  # hide download button
                     "Нет обработанных данных",  # progress message
-                    dedup_message  # dedup message
+                    dedup_message,  # dedup message
+                    None
                 )
                     
             except Exception as e:
@@ -822,16 +831,17 @@ def create_interface():
                     None,
                     gr.update(visible=False),
                     error_msg,
-                    ""
+                    "", 
+                    None
                 )
             finally:
                 if detector:
                     detector.cleanup()
         
-        def handle_download(file_bytes):
+        def handle_download(file_data):
             """Handle file download when button is clicked"""
-            if file_bytes is not None:
-                return (file_bytes, "результаты_анализа.xlsx")
+            if file_data is not None:
+                return file_data
             return None
             
         stop_btn.click(fn=stop_processing, outputs=[progress])
@@ -845,14 +855,15 @@ def create_interface():
                 events_plot,
                 download_button,
                 progress,
-                status_box
+                status_box, 
+                file_data
             ]
         )
         
         download_button.click(
             fn=handle_download,
-            inputs=[],
-            outputs=[gr.File(label="", visible=False)]
+            inputs=[file_data],
+            outputs=[file_output]
         )
         
     return app