v.1.08

app.py

@@ -88,42 +88,6 @@ class EventDetector:
             return "Negative"
         return "Neutral"
 
-def process_file(file):
-    try:
-        df = pd.read_excel(file, sheet_name='Публикации')
-        detector = EventDetector()
-        processed_rows = []
-        total = len(df)
-        
-        for idx, row in df.iterrows():
-            text = str(row.get('Выдержки из текста', ''))
-            entity = str(row.get('Объект', ''))
-            
-            event_type, event_summary = detector.detect_events(text, entity)
-            sentiment = detector.analyze_sentiment(text)
-            
-            processed_rows.append({
-                'Объект': entity,
-                'Заголовок': str(row.get('Заголовок', '')),
-                'Sentiment': sentiment,
-                'Event_Type': event_type,
-                'Event_Summary': event_summary,
-                'Текст': text
-            })
-            
-            # Log progress to console instead of updating UI directly
-            if idx % 10 == 0:
-                logger.info(f"Processed {idx}/{total} rows")
-        
-        result_df = pd.DataFrame(processed_rows)
-        logger.info("File processing complete!")
-        return result_df
-        
-    except Exception as e:
-        logger.error(f"File processing error: {e}")
-        gr.Error(f"Error processing file: {str(e)}")
-        return pd.DataFrame(columns=['Объект', 'Заголовок', 'Sentiment', 'Event_Type', 'Event_Summary', 'Текст'])
-
 def create_visualizations(df):
     if df is None or df.empty:
         return None, None
@@ -153,7 +117,7 @@ def create_visualizations(df):
 
 def create_interface():
     with gr.Blocks(theme=gr.themes.Soft()) as app:
-        gr.Markdown("# AI-анализ мониторинга новостей v.1.06")
+        gr.Markdown("# AI-анализ мониторинга новостей v.1.08")
         
         with gr.Row():
             file_input = gr.File(
@@ -188,14 +152,14 @@ def create_interface():
             with gr.Column():
                 events_plot = gr.Plot(label="Распределение событий")
                 
-        def analyze(file):
-            if file is None:
+        def analyze(file_bytes):
+            if file_bytes is None:
                 gr.Warning("Пожалуйста, загрузите файл")
                 return None, None, None, "Ожидание файла..."
             try:
-                # Instead of updating progress directly, we return new values
-                temp_file = file.name
-                df = process_file(temp_file)
+                # Convert bytes to BytesIO for pandas to read
+                file_obj = io.BytesIO(file_bytes)
+                df = process_file(file_obj)
                 
                 if df.empty:
                     return None, None, None, "Нет данных для обработки"
@@ -216,6 +180,42 @@ def create_interface():
         
     return app
 
+def process_file(file_obj):
+    try:
+        # Read Excel directly from BytesIO object
+        df = pd.read_excel(file_obj, sheet_name='Публикации')
+        detector = EventDetector()
+        processed_rows = []
+        total = len(df)
+        
+        for idx, row in df.iterrows():
+            text = str(row.get('Выдержки из текста', ''))
+            entity = str(row.get('Объект', ''))
+            
+            event_type, event_summary = detector.detect_events(text, entity)
+            sentiment = detector.analyze_sentiment(text)
+            
+            processed_rows.append({
+                'Объект': entity,
+                'Заголовок': str(row.get('Заголовок', '')),
+                'Sentiment': sentiment,
+                'Event_Type': event_type,
+                'Event_Summary': event_summary,
+                'Текст': text
+            })
+            
+            if idx % 10 == 0:
+                logger.info(f"Processed {idx}/{total} rows")
+        
+        result_df = pd.DataFrame(processed_rows)
+        logger.info("File processing complete!")
+        return result_df
+        
+    except Exception as e:
+        logger.error(f"File processing error: {e}")
+        gr.Error(f"Error processing file: {str(e)}")
+        return pd.DataFrame(columns=['Объект', 'Заголовок', 'Sentiment', 'Event_Type', 'Event_Summary', 'Текст'])
+
 if __name__ == "__main__":
     app = create_interface()
     app.launch(share=True)