Update app.py

app.py

@@ -1,7 +1,10 @@
 import streamlit as st
 import random
 import time
-
+import streamlit as st
+import random
+import time
+import os
 books = {
     "A Dictionary of Similes by Frank J. Wilstach": [
         "1. As strong as an ox.",
@@ -142,33 +145,15 @@ books = {
     "The American Heritage Thesaurus by Editors of the American Heritage Dictionaries": []
 }
 
-similes_metaphors = [simile for similes in books.values() for simile in similes]
-
-st.title("Similes and Metaphors Compendium")
-
-st.sidebar.title("Controls")
-if st.sidebar.button("Start Timer"):
-    start_time = time.time()
-    timer_active = True
-else:
-    timer_active = False
-
-if timer_active:
-    elapsed_time = int(time.time() - start_time)
-    st.sidebar.write(f"Timer: {elapsed_time} seconds")
 
-tab1, tab2 = st.tabs(["Random Similes and Metaphors", "Book List"])
-
-with tab1:
-    st.write("Randomly selected similes and metaphors:")
+def flatten_similes_metaphors(books):
+    return [simile for similes in books.values() for simile in similes]
 
+def display_random_similes_metaphors(similes_metaphors):
     display_lines = [""] * 5
-
     while True:
         random_samples = random.sample(similes_metaphors, 5)
-
         display_lines = [sample.split(". ", 1)[1] for sample in random_samples]
-
         simile_display = "\n".join([
             f"# {line}" if i == 0 else
             f"## {line}" if i == 1 else
@@ -178,16 +163,84 @@ with tab1:
             for i, line in enumerate(display_lines)
         ])
         st.markdown(simile_display)
-
         time.sleep(1)
 
-with tab2:
+def display_book_list(books):
     for book, samples in books.items():
         st.header(book)
         if samples:
-            st.write(f"Sample: {samples[0]}")
-            st.write("Additional Examples:")
-            for sample in samples[1:]:
-                st.write(f"- {sample}")
+            for i, sample in enumerate(samples):
+                if f"likes_{book}_{i}" not in st.session_state:
+                    st.session_state[f"likes_{book}_{i}"] = 0
+                col1, col2 = st.columns([9, 1])
+                with col1:
+                    st.write(f"- {sample}")
+                with col2:
+                    if st.button("👍", key=f"like_{book}_{i}"):
+                        st.session_state[f"likes_{book}_{i}"] += 1
+                    st.write(st.session_state[f"likes_{book}_{i}"])
+        else:
+            st.write("No similes or metaphors available for this book.")
+
+def handle_feedback(feedback):
+    if not os.path.exists("feedback"):
+        os.makedirs("feedback")
+    with open("feedback/feedback.md", "a") as f:
+        f.write(f"- {feedback}\n")
+
+def main():
+    books = {
+        # ... (books dictionary remains the same)
+    }
+
+    similes_metaphors = flatten_similes_metaphors(books)
+
+    st.title("Similes and Metaphors Compendium")
+
+    st.sidebar.title("Controls")
+    if st.sidebar.button("Start Timer"):
+        st.session_state.start_time = time.time()
+        st.session_state.timer_active = True
+    else:
+        st.session_state.timer_active = False
+
+    if st.session_state.timer_active:
+        elapsed_time = int(time.time() - st.session_state.start_time)
+        st.sidebar.write(f"Timer: {elapsed_time} seconds")
+
+    if "chat_history" not in st.session_state:
+        st.session_state.chat_history = []
+
+    user_input = st.text_input("Enter your message:")
+    if user_input:
+        st.session_state.chat_history.append(("user", user_input))
+        st.session_state.chat_history.append(("assistant", "I appreciate your message!"))
+
+    for i, (sender, message) in enumerate(st.session_state.chat_history):
+        if sender == "user":
+            st.chat_message(message, key=f"user_{i}", avatar_style="adventurer")
+            if st.button("👎", key=f"dislike_{i}"):
+                feedback = st.text_input("Please provide your feedback:", key=f"feedback_{i}")
+                if feedback:
+                    handle_feedback(feedback)
+                    st.markdown("Thank you for your feedback!")
+                    st.download_button(
+                        "Download Feedback",
+                        data=feedback,
+                        file_name="feedback.md",
+                        mime="text/markdown"
+                    )
         else:
-            st.write("No similes or metaphors available for this book.")
+            st.chat_message(message, key=f"assistant_{i}", avatar_style="persona")
+
+    tab1, tab2 = st.tabs(["Random Similes and Metaphors", "Book List"])
+
+    with tab1:
+        st.write("Randomly selected similes and metaphors:")
+        display_random_similes_metaphors(similes_metaphors)
+
+    with tab2:
+        display_book_list(books)
+
+if __name__ == "__main__":
+    main()