updateapp

.gitignore

@@ -1 +1,2 @@
-**.env
+**.env
+embeddings_db/

app.py

@@ -10,6 +10,17 @@ st.set_page_config(
 st.markdown(
     """
     <style>
+    /* Force light mode */
+    [data-testid="stAppViewContainer"] {
+        background-color: #f8f9fa;
+    }
+    [data-testid="stSidebar"] {
+        background-color: #f8f9fa;
+    }
+    .stApp {
+        background-color: #f8f9fa;
+        color: #000000;
+    }
     .main {
         background-color: #f8f9fa;
     }
@@ -33,13 +44,28 @@ st.markdown(
         padding: 2px 5px;
         border-radius: 3px;
         font-weight: bold;
-        color: #856404;
+        color: #000000;
     }
     .answer-text {
         color: #000000;
         font-size: 16px;
         line-height: 1.6;
     }
+    .section-title {
+        color: #000000;
+        font-weight: bold;
+    }
+    .section-text {
+        color: #000000;
+    }
+    /* Override any dark mode styles */
+    .stMarkdown {
+        color: #000000;
+    }
+    .stTextInput>div>div>input {
+        background-color: #ffffff;
+        color: #000000;
+    }
     </style>
 """,
     unsafe_allow_html=True,
@@ -52,8 +78,8 @@ st.title("🇺🇸 Cory Booker's Historic Filibuster RAG")
 st.markdown(
     """
     <div style='background-color: #ffffff; padding: 20px; border-radius: 10px; box-shadow: 0 2px 4px rgba(0,0,0,0.1); margin-top: 20px;'>
-        <h3 style='color: #1a237e;'>Source Material</h3>
-        <p>The transcript is derived from videos in this playlist: 
+        <h3 class="section-title">Source Material</h3>
+        <p class="section-text">The transcript is derived from videos in this playlist: 
         <a href='https://www.youtube.com/playlist?list=PLeifkaZBt4JtdT8DZ7aftJ0lU0Q6Hfnvz' target='_blank'>YouTube Playlist</a></p>
     </div>
 """,
@@ -63,8 +89,8 @@ st.markdown(
 st.markdown(
     """
     <div style='background-color: #ffffff; padding: 20px; border-radius: 10px; box-shadow: 0 2px 4px rgba(0,0,0,0.1); margin-top: 20px;'>
-        <h2 style='color: #1a237e;'>Explore the Filibuster</h2>
-        <p>Ask questions about Senator Booker's historic <span class="highlight-text">25-hour filibuster</span> speech below:</p>
+        <h2 class="section-title">Explore the Filibuster</h2>
+        <p class="section-text">Ask questions about Senator Booker's historic <span class="highlight-text">25-hour filibuster</span> speech below:</p>
     </div>
 """,
     unsafe_allow_html=True,
@@ -82,7 +108,7 @@ if st.button("Search", key="search_button"):
         st.markdown(
             """
             <div style='background-color: #ffffff; padding: 20px; border-radius: 10px; box-shadow: 0 2px 4px rgba(0,0,0,0.1); margin-top: 20px;'>
-                <h3 style='color: #1a237e;'>Answer:</h3>
+                <h3 class="section-title">Answer:</h3>
                 <p class="answer-text">{}</p>
             </div>
         """.format(
@@ -91,12 +117,22 @@ if st.button("Search", key="search_button"):
             unsafe_allow_html=True,
         )
 
+        # Add disclaimer about LLM hallucinations
+        st.markdown(
+            """
+            <div style='background-color: #fff3cd; padding: 15px; border-radius: 5px; margin-top: 20px; border: 1px solid #ffeeba;'>
+                <p style='color: #856404; margin: 0;'>Please verify important facts and consult primary sources when possible.</p>
+            </div>
+        """,
+            unsafe_allow_html=True,
+        )
+
 # Footer
 st.markdown(
     """
-    <div style='text-align: center; margin-top: 40px; color: #666;'>
-        <p>🇺🇸 Celebrating American Democracy and the Power of Speech 🇺🇸</p>
-        <p style='font-size: 14px;'>A tribute to Senator Cory Booker's record-breaking <span class="highlight-text">25-hour filibuster</span></p>
+    <div style='text-align: center; margin-top: 40px;'>
+        <p class="section-text">🇺🇸 Celebrating American Democracy and the Power of Speech 🇺🇸</p>
+        <p class="section-text" style='font-size:   14px;'>Made with ❤️ by <a href="https://nkasmanoff.github.io/" target="_blank">Noah K</a></p>
     </div>
 """,
     unsafe_allow_html=True,

rag.py

@@ -36,14 +36,12 @@ class COT_RAG(dspy.Module):
         self.respond = dspy.ChainOfThought("context, question -> response")
 
     def forward(self, question):
-        question_ = (
-            "Please review this speech by Cory Booker and answer the question below."
-        )
-        question_ += f"\n\nQuestion: {question}"
-        docs = retrieve({"question": question_})
+        question_ = "Given the context from Cory Booker's speech, please answer the question below."
+        question_ += f"\n\nQuestion: {question}\n\nStart your answer by specifying this was from Senator Booker."
+        docs = retrieve({"question": question})
         self.docs = docs
         context = [doc.page_content for doc in docs]
-        return self.respond(context=context, question=question)
+        return self.respond(context=context, question=question_)
 
 
 def answer_question(question):