v.1.55

app.py

@@ -230,24 +230,23 @@ class EventDetector:
             return "Neutral"
     
     def estimate_impact(self, text, entity):
-        """Estimate impact using Groq for negative sentiment texts"""
+        """Estimate impact using Groq ensuring sentiment-impact alignment"""
         try:
             if not self.groq:
                 return "Неопределенный эффект", "Groq API недоступен"
                 
             template = """
-            You are a financial analyst. Analyze this news about {entity} and assess its potential impact.
+            You are a financial risk analyst. Analyze this negative news about {entity}:
             
             News: {news}
             
-            Classify the impact into one of these categories:
-            1. "Значительный риск убытков" (Significant loss risk)
-            2. "Умеренный риск убытков" (Moderate loss risk)
-            3. "Незначительный риск убытков" (Minor loss risk)
-            4. "Вероятность прибыли" (Potential profit)
-            5. "Неопределенный эффект" (Uncertain effect)
+            Based on the severity of negative implications, classify the impact as:
+            1. "Значительный риск убытков" - for severe negative events with high financial impact
+            2. "Умеренный риск убытков" - for moderate negative events with medium financial impact
+            3. "Незначительный риск убытков" - for minor negative events with low financial impact
+            4. "Неопределенный эффект" - if impact cannot be reliably assessed
             
-            Format your response exactly as:
+            Format response exactly as:
             Impact: [category]
             Reasoning: [explanation in 2-3 sentences]
             """
@@ -267,6 +266,15 @@ class EventDetector:
                 parts = response_text.split("Reasoning:")
                 impact = parts[0].split("Impact:")[1].strip()
                 reasoning = parts[1].strip()
+                
+                # Validate impact category
+                valid_impacts = {
+                    "Значительный риск убытков",
+                    "Умеренный риск убытков", 
+                    "Незначительный риск убытков",
+                    "Неопределенный эффект"
+                }
+                impact = impact if impact in valid_impacts else "Неопределенный эффект"
             else:
                 impact = "Неопределенный эффект"
                 reasoning = "Не удалось определить влияние"
@@ -279,32 +287,28 @@ class EventDetector:
 
     @spaces.GPU(duration=60)
     def process_text(self, text, entity):
-        """Process text through translation, sentiment, and impact analysis"""
+        """Process text with Groq-driven sentiment analysis"""
         try:
-            # Translate text
             translated_text = self._translate_text(text)
+            initial_sentiment = self.analyze_sentiment(translated_text)
             
-            # Analyze sentiment
-            sentiment = self.analyze_sentiment(translated_text)
-            
-            # Initialize impact and reasoning
             impact = "Неопределенный эффект"
             reasoning = ""
             
-            # If sentiment is negative, estimate impact and produce impact reasoning translated back into russian
-            if sentiment == "Negative":
-                
-                impact, reasoning = self.estimate_impact(translated_text, entity)
-                reasoning = self.rutranslator(reasoning)[0]['translation_text']
+            # Always get Groq analysis for all texts
+            impact, reasoning = self.estimate_impact(translated_text, entity)
+            reasoning = self.rutranslator(reasoning)[0]['translation_text']
             
-            # Detect events
-            event_type, event_summary = self.detect_events(translated_text, entity)
-
-            #translate Reasoning back 
+            # Override sentiment based on Groq impact
+            final_sentiment = initial_sentiment
+            if impact == "Вероятность прибыли":
+                final_sentiment = "Positive"
+            
+            event_type, event_summary = self.detect_events(text, entity)
                                     
             return {
                 'translated_text': translated_text,
-                'sentiment': sentiment,
+                'sentiment': final_sentiment,
                 'impact': impact,
                 'reasoning': reasoning,
                 'event_type': event_type,
@@ -316,7 +320,7 @@ class EventDetector:
             return {
                 'translated_text': '',
                 'sentiment': 'Neutral',
-                'impact': 'Неопределенный эффект',
+                'impact': 'Неопределенный эффект', 
                 'reasoning': f'Ошибка обработки: {str(e)}',
                 'event_type': 'Нет',
                 'event_summary': ''
@@ -653,7 +657,7 @@ def create_interface():
         # Create state for file data
         current_file = gr.State(None)
         
-        gr.Markdown("# AI-анализ мониторинга новостей v.1.54")
+        gr.Markdown("# AI-анализ мониторинга новостей v.1.55")
         
         with gr.Row():
             file_input = gr.File(