Update main.py

main.py

@@ -1,11 +1,11 @@
 from flask import Flask, render_template, request, send_file
 from services.aggregator import collect_data
 from services.sentiment import predict
+from services.evaluation import evaluate_model
 from collections import Counter
 import pandas as pd
 from wordcloud import WordCloud
-import matplotlib.pyplot as plt
-from sklearn.metrics import classification_report
+import os
 
 app = Flask(__name__)
 
@@ -24,22 +24,22 @@ def index():
 
         counts = Counter(sentiments)
 
-        # 📊 PER PLATFORM
+        # 📊 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
+        # ☁️ wordcloud
         try:
-            text_all = " ".join(texts)
-            wc = WordCloud(width=800, height=400).generate(text_all)
+            os.makedirs("static", exist_ok=True)
+            wc = WordCloud(width=800, height=400).generate(" ".join(texts))
             wc.to_file("static/wordcloud.png")
         except:
             pass
 
-        # 📁 SAVE CSV
+        # 📁 CSV
         df = pd.DataFrame({
             "text": texts,
             "sentiment": sentiments,
@@ -47,9 +47,8 @@ def index():
         })
         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)
+        # 📊 evaluasi
+        eval_result = evaluate_model(predict)
 
         data = list(zip(texts, sentiments, sources))
 
@@ -58,9 +57,9 @@ def index():
             data=data,
             counts=counts,
             platform_counts=platform_counts,
-            report=report,
-            source=source,
-            keyword=keyword
+            eval_result=eval_result,
+            keyword=keyword,
+            source=source
         )
 
     return render_template("index.html")
@@ -68,4 +67,8 @@ def index():
 
 @app.route('/download')
 def download():
-    return send_file("static/result.csv", as_attachment=True)
+    return send_file("static/result.csv", as_attachment=True)
+
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=7860)