Update main.py

main.py

@@ -8,16 +8,25 @@ app = Flask(__name__)
 @app.route('/', methods=['GET', 'POST'])
 def index():
     if request.method == 'POST':
-        keyword = request.form['keyword']
-        source = request.form['source']
+        try:
+            keyword = request.form.get('keyword', '')
+            source = request.form.get('source', 'all')
 
-        comments = collect_data(keyword, source)
-        sentiments = predict(comments[:100])
+            comments = collect_data(keyword, source)
 
-        counts = Counter(sentiments)
-        data = list(zip(comments[:100], sentiments))
+            if not comments:
+                return render_template("result.html", data=[], counts={}, source=source)
 
-        return render_template("result.html", data=data, counts=counts, source=source)
+            sentiments = predict(comments[:50])  # 🔥 limit biar ringan
+
+            counts = Counter(sentiments)
+            data = list(zip(comments[:50], sentiments))
+
+            return render_template("result.html", data=data, counts=counts, source=source)
+
+        except Exception as e:
+            print("ERROR:", e)
+            return render_template("result.html", data=[], counts={}, source="error")
 
     return render_template("index.html")