Update main.py

main.py

@@ -9,15 +9,15 @@ app = Flask(__name__)
 def index():
     if request.method == 'POST':
         keyword = request.form['keyword']
+        source = request.form['source']
 
-        comments = collect_data(keyword)
+        comments = collect_data(keyword, source)
         sentiments = predict(comments[:100])
 
         counts = Counter(sentiments)
-
         data = list(zip(comments[:100], sentiments))
 
-        return render_template("result.html", data=data, counts=counts)
+        return render_template("result.html", data=data, counts=counts, source=source)
 
     return render_template("index.html")