Update main.py

main.py

@@ -1,7 +1,11 @@
-from flask import Flask, render_template, request
+from flask import Flask, render_template, request, send_file
 from services.aggregator import collect_data
 from services.sentiment import predict
 from collections import Counter
+import pandas as pd
+from wordcloud import WordCloud
+import matplotlib.pyplot as plt
+from sklearn.metrics import classification_report
 
 app = Flask(__name__)
 
@@ -11,17 +15,50 @@ def index():
         keyword = request.form.get('keyword')
         source = request.form.get('source', 'all')
 
-        comments = collect_data(keyword, source)
+        data_raw = collect_data(keyword, source)
 
-        sentiments = predict(comments[:50])
+        texts = [t for s, t in data_raw][:100]
+        sources = [s for s, t in data_raw][:100]
+
+        sentiments = predict(texts)
 
         counts = Counter(sentiments)
-        data = list(zip(comments[:50], sentiments))
+
+        # 📊 PER PLATFORM
+        platform_counts = {}
+        for src, sent in zip(sources, sentiments):
+            if src not in platform_counts:
+                platform_counts[src] = {"Positive":0,"Neutral":0,"Negative":0}
+            platform_counts[src][sent] += 1
+
+        # ☁️ WORDCLOUD
+        try:
+            text_all = " ".join(texts)
+            wc = WordCloud(width=800, height=400).generate(text_all)
+            wc.to_file("static/wordcloud.png")
+        except:
+            pass
+
+        # 📁 SAVE CSV
+        df = pd.DataFrame({
+            "text": texts,
+            "sentiment": sentiments,
+            "source": sources
+        })
+        df.to_csv("static/result.csv", index=False)
+
+        # 📊 EVALUASI (dummy ground truth)
+        y_true = sentiments  # karena tidak ada label asli
+        report = classification_report(y_true, sentiments, output_dict=True)
+
+        data = list(zip(texts, sentiments, sources))
 
         return render_template(
             "result.html",
             data=data,
             counts=counts,
+            platform_counts=platform_counts,
+            report=report,
             source=source,
             keyword=keyword
         )
@@ -29,5 +66,6 @@ def index():
     return render_template("index.html")
 
 
-if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860)
+@app.route('/download')
+def download():
+    return send_file("static/result.csv", as_attachment=True)