Update app.py

app.py

@@ -4,7 +4,7 @@ import pandas as pd
 import json
 import os
 
-def display_table():
+def display_table(vote_data):
     st.title("The Great Debate: Vote on the Funniest Questions!")
 
     data = [
@@ -20,7 +20,16 @@ def display_table():
         (10, "🙃", "What's the most useless talent you have?", "[Wikipedia](https://en.wikipedia.org/wiki/Talent_(human))"),
     ]
 
-    st.table(pd.DataFrame(data, columns=['Question', 'Emoji', 'Title', 'Description']))
+    table_data = []
+    for row in data:
+        upvotes, downvotes = count_votes(vote_data, f"Question {row[0]}")
+        table_data.append(row + (upvotes, downvotes))
+
+    df = pd.DataFrame(table_data, columns=['Question', 'Emoji', 'Title', 'Description', 'Upvotes', 'Downvotes'])
+
+    st.table(df)
+
+    return df
 
 def update_vote_log(term, vote_type):
     with open('vote.log.txt', 'a') as f:
@@ -42,41 +51,21 @@ def count_votes(vote_data, term):
     return upvotes, downvotes
 
 def main():
-    display_table()
-
-    terms = [
-        'Question 1',
-        'Question 2',
-        'Question 3',
-        'Question 4',
-        'Question 5',
-        'Question 6',
-        'Question 7',
-        'Question 8',
-        'Question 9',
-        'Question 10'
-    ]
-
     vote_data = load_vote_log()
 
-    with st.sidebar:
-        st.title("User Controls")
-        for term in terms:
-            st.write(term)
-            upvotes, downvotes = count_votes(vote_data, term)
-            st.write(f"Total upvotes: {upvotes}")
-            st.write(f"Total downvotes: {downvotes}")
-            
-        upvote_button = st.button(f"👍 Upvote {term}")
-        downvote_button = st.button(f"👎 Downvote {term}")
-
-        if upvote_button:
-            update_vote_log(term, 'upvote')
-            st.experimental_rerun()
+    df = display_table(vote_data)
 
-        if downvote_button:
-            update_vote_log(term, 'downvote')
-            st.experimental_rerun()
+    for index, row in df.iterrows():
+        question_id = f"Question {row['Question']}"
+        upvote_button = st.button(f"👍 Upvote {question_id}")
+        downvote_button = st.button(f"👎 Downvote {question_id}")
+    if upvote_button:
+        update_vote_log(question_id, 'upvote')
+        st.experimental_rerun()
 
-main()
+    if downvote_button:
+        update_vote_log(question_id, 'downvote')
+        st.experimental_rerun()
 
+    
+main()