Fix HF Spaces deployment: Add __init__.py and simplify app.py for better compatibility

app.py

@@ -1,59 +1,70 @@
 import gradio as gr
-import sys
 import os
+import sys
 
-# Add the current directory to Python path
-sys.path.append(os.path.dirname(os.path.abspath(__file__)))
+# Add current directory to path
+current_dir = os.path.dirname(os.path.abspath(__file__))
+sys.path.insert(0, current_dir)
 
-# Import your advanced model system
+# Try to import advanced model, fallback to basic if needed
 try:
-    from app.advanced_model import predict_advanced, get_advanced_analyzer
-    ADVANCED_AVAILABLE = True
-except ImportError:
-    # Fallback to basic model if advanced isn't available
     from app.model import predict
-    ADVANCED_AVAILABLE = False
+    from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification
+    
+    # Try to load your custom model first
+    try:
+        MODEL_NAME = "fitsblb/YelpReviewsAnalyzer"
+        tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
+        model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME)
+        sentiment_pipeline = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
+        CUSTOM_MODEL_AVAILABLE = True
+    except:
+        # Fallback to a general model
+        sentiment_pipeline = pipeline("sentiment-analysis", model="cardiffnlp/twitter-roberta-base-sentiment-latest")
+        CUSTOM_MODEL_AVAILABLE = False
+        
+except ImportError:
+    # Ultimate fallback
+    from transformers import pipeline
+    sentiment_pipeline = pipeline("sentiment-analysis", model="cardiffnlp/twitter-roberta-base-sentiment-latest")
+    CUSTOM_MODEL_AVAILABLE = False
 
 def analyze_sentiment(text):
-    """Analyze sentiment using the advanced multi-model system"""
+    """Analyze sentiment using available models"""
     if not text.strip():
         return "Please enter some text to analyze!"
     
     try:
-        if ADVANCED_AVAILABLE:
-            # Use advanced multi-model system
-            result = predict_advanced(text)
-            
-            # Format results for display
-            model_results = []
-            for model_result in result.results:
-                model_results.append(f"**{model_result.model_name}**: {model_result.sentiment} ({model_result.confidence:.3f})")
+        # Use the pipeline
+        result = sentiment_pipeline(text)
+        
+        if isinstance(result, list) and len(result) > 0:
+            result = result[0]
+        
+        sentiment = result['label']
+        confidence = result['score']
+        
+        # Map labels to consistent format
+        if sentiment.upper() in ['POSITIVE', 'POS']:
+            sentiment = "Positive"
+        elif sentiment.upper() in ['NEGATIVE', 'NEG']:
+            sentiment = "Negative"
+        elif sentiment.upper() in ['NEUTRAL', 'NEU']:
+            sentiment = "Neutral"
             
-            output = f"""
-## 🎯 Consensus Result
-**Sentiment**: {result.consensus_sentiment}  
-**Confidence**: {result.average_confidence:.3f}  
-**Agreement Score**: {result.agreement_score:.3f}  
-**Processing Time**: {result.processing_time:.3f}s
-
-## 🤖 Individual Model Results
-{chr(10).join(model_results)}
-
----
-*Powered by 4 AI models working together for superior accuracy!*
-            """
-            return output
-        else:
-            # Fallback to basic model
-            sentiment, confidence = predict(text)
-            return f"""
-## 📊 Sentiment Analysis Result
+        model_info = "YelpReviewsAnalyzer (Custom)" if CUSTOM_MODEL_AVAILABLE else "RoBERTa (Fallback)"
+        
+        output = f"""
+## 🎯 Sentiment Analysis Result
 **Sentiment**: {sentiment}  
-**Confidence**: {confidence:.3f}
+**Confidence**: {confidence:.3f}  
+**Model**: {model_info}
 
-*Using YelpReviewsAnalyzer model*
-            """
-    
+---
+*Analyzing sentiment with AI models*
+        """
+        return output
+        
     except Exception as e:
         return f"❌ Error analyzing sentiment: {str(e)}"
 
@@ -66,17 +77,12 @@ demo = gr.Interface(
         lines=3
     ),
     outputs=gr.Markdown(label="🎯 Analysis Results"),
-    title="🚀 Advanced Sentiment Analyzer",
+    title="🚀 Sentiment Analyzer",
     description="""
-    **Multi-Model AI System for Superior Sentiment Analysis**
-    
-    This system uses up to 4 different AI models working together to provide more accurate sentiment predictions:
-    - 🎯 YelpReviewsAnalyzer (custom fine-tuned model)
-    - 🤖 DistilBERT (general-purpose)
-    - 🐦 Twitter-RoBERTa (social media optimized)  
-    - 💰 FinBERT (financial sentiment)
+    **AI-Powered Sentiment Analysis**
     
-    The models vote on the final prediction using a consensus algorithm for higher accuracy!
+    This system analyzes the sentiment of your text using transformer models.
+    Enter any text and get instant sentiment predictions with confidence scores!
     """,
     examples=[
         ["This restaurant has absolutely amazing food and incredible service!"],

app/__init__.py

@@ -0,0 +1 @@
+# Empty file to make this directory a Python package