Update src/streamlit_app.py

src/streamlit_app.py

@@ -77,6 +77,22 @@ llm_client = load_llm_client()
 if llm_client:
     st.success("✅ GitHub Models API connected automatically!")
 
+def fix_problematic_words(text):
+    """Quick fix for specific words causing classification issues"""
+    fixes = {
+        'phenomenal': 'absolutely amazing excellent',
+        'fabulous': 'excellent wonderful', 
+        'superb': 'excellent great quality',
+        'marvelous': 'amazing wonderful',
+        'spectacular': 'outstanding excellent',
+    }
+    
+    text_fixed = text
+    for problematic, replacement in fixes.items():
+        text_fixed = re.sub(rf'\b{problematic}\b', replacement, text_fixed, flags=re.IGNORECASE)
+    
+    return text_fixed
+
 def predict_sentiment_enhanced(text):
     """Enhanced sentiment prediction ensuring 5-class output"""
     if not text.strip():
@@ -89,7 +105,7 @@ def predict_sentiment_enhanced(text):
         }
     
     try:
-        results = sentiment_pipeline(text)
+        results = sentiment_pipeline(fix_problematic_words(text))
         if isinstance(results[0], list):
             results = results[0]