Update main.py

main.py

@@ -8,25 +8,23 @@ app = Flask(__name__)
 @app.route('/', methods=['GET', 'POST'])
 def index():
     if request.method == 'POST':
-        try:
-            keyword = request.form.get('keyword', '')
-            source = request.form.get('source', 'all')
+        keyword = request.form.get('keyword')
+        source = request.form.get('source', 'all')
 
-            comments = collect_data(keyword, source)
+        comments = collect_data(keyword, source)
 
-            if not comments:
-                return render_template("result.html", data=[], counts={}, source=source)
+        sentiments = predict(comments[:50])
 
-            sentiments = predict(comments[:50])  # 🔥 limit biar ringan
+        counts = Counter(sentiments)
+        data = list(zip(comments[:50], sentiments))
 
-            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(
+            "result.html",
+            data=data,
+            counts=counts,
+            source=source,
+            keyword=keyword
+        )
 
     return render_template("index.html")