Initial commit for HF Spaces deployment

Dockerfile

@@ -0,0 +1,15 @@
+FROM python:3.10
+
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+
+WORKDIR /app
+
+COPY --chown=user ./requirements.txt requirements.txt
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+
+COPY --chown=user . /app
+
+# For Flask, we should use gunicorn or similar, not uvicorn (which is for FastAPI)
+CMD ["gunicorn", "-b", "0.0.0.0:7860", "app:app"]

app.py

@@ -0,0 +1,117 @@
+import matplotlib
+matplotlib.use('Agg')
+
+from flask import Flask, render_template, request, redirect, url_for
+import torch
+import pandas as pd
+import matplotlib.pyplot as plt
+import io
+import base64
+from wordcloud import WordCloud
+from transformers import DistilBertTokenizer, DistilBertForSequenceClassification
+from datasets import load_dataset
+
+app = Flask(__name__)
+
+# Global variables
+product_data = None
+tokenizer = None
+model = None
+
+def load_data_and_model():
+    global product_data, tokenizer, model
+    
+    # Load dataset
+    dataset = load_dataset("LoganKells/amazon_product_reviews_video_games")
+    product_data = dataset["train"].to_pandas()
+    
+    # Load model and tokenizer
+    tokenizer = DistilBertTokenizer.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
+    model = DistilBertForSequenceClassification.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
+
+def generate_word_cloud(text):
+    if not text.strip():
+        return None
+        
+    wordcloud = WordCloud(width=800, height=400, background_color='white').generate(text)
+    plt.figure(figsize=(8, 4))
+    plt.imshow(wordcloud, interpolation='bilinear')
+    plt.axis('off')
+    
+    img_buffer = io.BytesIO()
+    plt.savefig(img_buffer, format='png')
+    img_buffer.seek(0)
+    img_base64 = base64.b64encode(img_buffer.getvalue()).decode('utf-8')
+    plt.close()
+    
+    return img_base64
+
+@app.route('/', methods=['GET', 'POST'])
+def index():
+    global product_data, tokenizer, model
+    
+    # Load data and model if not already loaded
+    if product_data is None or tokenizer is None or model is None:
+        load_data_and_model()
+    
+    result = None
+    error = None
+    
+    if request.method == 'POST':
+        if 'refresh' in request.form:
+            return redirect(url_for('index'))
+            
+        if 'generate' in request.form:
+            product_code = request.form.get('product_code', '').strip()
+            
+            # Validate product code (10 digits)
+            if not product_code or not product_code.isdigit() or len(product_code) != 10:
+                error = "Please enter a 10 digit number product code."
+            else:
+                # Get reviews for the product
+                product_reviews = product_data[product_data['asin'] == product_code]['reviewText'].tolist()
+                
+                if not product_reviews:
+                    error = "Product code not found."
+                else:
+                    total_reviews = len(product_reviews)
+                    positive_count = 0
+                    negative_count = 0
+                    positive_reviews_text = ""
+                    negative_reviews_text = ""
+
+                    # Perform sentiment analysis on each review
+                    for review in product_reviews:
+                        inputs = tokenizer(review, return_tensors="pt", truncation=True, padding=True)
+                        with torch.no_grad():
+                            logits = model(**inputs).logits
+                        predicted_class_id = logits.argmax().item()
+                        label = model.config.id2label[predicted_class_id]
+                        
+                        if label == 'POSITIVE':
+                            positive_count += 1
+                            positive_reviews_text += review + " "
+                        else:
+                            negative_count += 1
+                            negative_reviews_text += review + " "
+
+                    positive_percentage = (positive_count / total_reviews) * 100
+                    negative_percentage = (negative_count / total_reviews) * 100
+
+                    # Generate word clouds
+                    positive_wordcloud = generate_word_cloud(positive_reviews_text)
+                    negative_wordcloud = generate_word_cloud(negative_reviews_text)
+
+                    result = {
+                        'product_code': product_code,
+                        'total_reviews': total_reviews,
+                        'positive_percentage': positive_percentage,
+                        'negative_percentage': negative_percentage,
+                        'positive_wordcloud': positive_wordcloud,
+                        'negative_wordcloud': negative_wordcloud
+                    }
+    
+    return render_template('index.html', result=result, error=error)
+
+if __name__ == '__main__':
+    app.run(debug=True)

templates/index.html

@@ -0,0 +1,140 @@
+<!DOCTYPE html>
+<html lang="en">
+  <head>
+    <meta charset="UTF-8" />
+    <meta name="viewport" content="width=device-width, initial-scale=1.0" />
+    <title>Product Health Check</title>
+    <style>
+      body {
+        font-family: Arial, sans-serif;
+        max-width: 1200px;
+        margin: 0 auto;
+        padding: 20px;
+      }
+      h1 {
+        text-align: center;
+        margin-bottom: 30px;
+      }
+      .form-container {
+        display: flex;
+        flex-direction: column;
+        align-items: center;
+        margin-bottom: 30px;
+      }
+      .input-row {
+        display: flex;
+        width: 100%;
+        max-width: 600px;
+        margin-top: 10px;
+      }
+      .input-label {
+        display: block;
+        margin-bottom: 5px;
+        font-weight: bold;
+      }
+      input[type="text"] {
+        flex-grow: 1;
+        padding: 8px;
+        font-size: 16px;
+        border: 1px solid #ccc;
+        border-radius: 4px;
+      }
+      button {
+        margin-left: 10px;
+        padding: 8px 16px;
+        font-size: 16px;
+        background-color: #4caf50;
+        color: white;
+        border: none;
+        border-radius: 4px;
+        cursor: pointer;
+      }
+      button[name="refresh"] {
+        background-color: #f44336;
+      }
+      .error {
+        color: red;
+        margin: 15px 0;
+        text-align: center;
+      }
+      .results {
+        margin-top: 20px;
+      }
+      .results-header {
+        margin-bottom: 20px;
+      }
+      .results-columns {
+        display: flex;
+        gap: 20px;
+      }
+      .column {
+        flex: 1;
+        border: 1px solid #ddd;
+        border-radius: 4px;
+        padding: 15px;
+      }
+      .column h3 {
+        margin-top: 0;
+      }
+      img {
+        max-width: 100%;
+        height: auto;
+      }
+    </style>
+  </head>
+  <body>
+    <h1>Check Your Products Health</h1>
+
+    <div class="form-container">
+      <div class="input-label">
+        Enter Product Code (ASIN): ASIN is a 10 digit number
+      </div>
+      <form method="POST" class="input-row">
+        <input
+          type="text"
+          name="product_code"
+          value="{{ request.form.get('product_code', '') }}"
+        />
+        <button type="submit" name="generate" value="true">Generate</button>
+        <button type="submit" name="refresh" value="true">Refresh</button>
+      </form>
+    </div>
+
+    {% if error %}
+    <div class="error">{{ error }}</div>
+    {% endif %} {% if result %}
+    <div class="results">
+      <div class="results-header">
+        <h2>Product ID: {{ result.product_code }}</h2>
+        <p>Total Reviews: {{ result.total_reviews }}</p>
+      </div>
+
+      <div class="results-columns">
+        <div class="column">
+          <h3>
+            Positive Reviews: {{ "%.2f"|format(result.positive_percentage) }}%
+          </h3>
+          {% if result.positive_wordcloud %}
+          <img
+            src="data:image/png;base64,{{ result.positive_wordcloud }}"
+            alt="Positive Reviews Word Cloud"
+          />
+          {% endif %}
+        </div>
+
+        <div class="column">
+          <h3>
+            Negative Reviews: {{ "%.2f"|format(result.negative_percentage) }}%
+          </h3>
+          {% if result.negative_wordcloud %}
+          <img
+            src="data:image/png;base64,{{ result.negative_wordcloud }}"
+            alt="Negative Reviews Word Cloud"
+          />
+          {% endif %}
+        </div>
+      </div>
+    </div>
+    {% endif %}
+  </body>
+</html>