Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,4 +1,5 @@
 import os
+os.environ["STREAMLIT_HOME"] = "/tmp/.streamlit"
 from langchain_huggingface import HuggingFaceEndpoint
 import streamlit as st
 
@@ -18,23 +19,24 @@ if "response" not in st.session_state:
     st.session_state.response = ""
 
 def get_llm(model_id=MODEL, max_new_tokens=300, temperature=0.7):
-    os.environ["HF_TOKEN"] = hf_token  # ensure token is accessible to endpoint
+    os.environ["HF_TOKEN"] = os.getenv("HF_TOKEN")  # Optional but ensures it's set
+
     return HuggingFaceEndpoint(
         repo_id=model_id,
         max_new_tokens=max_new_tokens,
         temperature=temperature,
     )
-
+    
 # create llm
 llm = get_llm()
 
-# prompts
+#  prompts
 prompt = f"""
     You are an AI assistant designed to support high school students in the subject of {SUBJECT}. 
     Your role is to offer friendly, helpful, concise, in-depth guidance, just like a supportive teacher would.
     
     Please follow these guidelines:
-
+    
     1. Maintain a polite, respectful, and professional tone at all times.
     2. Adhere to ethical principles — do not promote cheating, harmful behavior, or misinformation.
     3. Interact in a warm, encouraging, and student-centered style — use clear explanations, positive reinforcement, and examples when needed.
@@ -54,10 +56,11 @@ p_application = """
     6. Do not include any explanation or examples in your response.
 """
 
-# count the number of times "I don't know" is clicked
+# count the number of times "I don't know is clicked"
 if "retry_count" not in st.session_state:
     st.session_state.retry_count = 0
 
+    
 # Initialize session state
 if "help_clicks" not in st.session_state:
     st.session_state.help_clicks = 0
@@ -72,8 +75,9 @@ st.markdown(
     f"<p style='font-size:20px;'>{QUESTION}</p>",
     unsafe_allow_html=True
 )
-
+# Outer container for neat padding
 with st.container():
+    # Question area
     st.text_area(
         label="Type your response here.",
         value="",
@@ -83,74 +87,87 @@ with st.container():
 
     st.markdown("")
 
+    # Help Button Logic
     def toggle_help():
         st.session_state.help_clicks += 1
-        st.session_state.button_clicked = None
+        st.session_state.button_clicked = None  # Reset help text on new toggle
 
+    # Help Button (main toggle)
     col1, col2, col3 = st.columns([1, 3, 1])
     with col2:
         st.button("Help", on_click=toggle_help)
 
+    # Show 3 sub-buttons if Help clicked an odd number of times
     if st.session_state.help_clicks % 2 == 1:
         st.markdown("### Need Help?")
         st.markdown("Choose an option below to better understand the question.")
         with st.container():
-            st.markdown("---")
+            st.markdown("---")  # Divider for clarity
             col1, col2, col3 = st.columns(3)
 
             with col1:
                 if st.button("📝 Explain the question"):
                     if st.session_state.button_clicked != "Explain the question":
-                        full_prompt = f"[INST]<<SYS>>\n{prompt + p_explanation}\n<</SYS>>\n\n{QUESTION}\n[/INST]"
+                        # First time clicked
+                        full_prompt = (
+                            "[INST]<<SYS>>\n"
+                            f"{prompt + p_explanation}\n"
+                            "<</SYS>>\n\n"
+                            f"{QUESTION}\n"
+                            "[/INST]"
+                        )
                         st.session_state.response = llm.invoke(full_prompt)
                         st.session_state.retry_count = 0
-                        st.session_state.full_prompt = full_prompt
+                        st.session_state.full_prompt = full_prompt  # Save prompt for retry
                         st.session_state.button_clicked = "Explain the question"
             with col2:
-                if st.button("💡 Give an example"):
-                    if st.session_state.button_clicked != "Give an example":
-                        full_prompt = f"[INST]<<SYS>>\n{prompt + p_example}\n<</SYS>>\n\n{QUESTION}\n[/INST]"
-                        st.session_state.response = llm.invoke(full_prompt)
-                        st.session_state.retry_count = 0
-                        st.session_state.full_prompt = full_prompt
-                        st.session_state.button_clicked = "Give an example"
+                    if st.button("💡 Give an example"):
+                        if st.session_state.button_clicked != "Give an example":
+                            # First time clicked
+                            full_prompt = (
+                                "[INST]<<SYS>>\n"
+                                f"{prompt + p_example}\n"
+                                "<</SYS>>\n\n"
+                                f"{QUESTION}\n"
+                                "[/INST]"
+                            )
+                            st.session_state.response = llm.invoke(full_prompt)
+                            st.session_state.retry_count = 0
+                            st.session_state.full_prompt = full_prompt  # Save prompt for retry
+                            st.session_state.button_clicked = "Give an example"
             with col3:
                 if st.button("🤔 Who cares?"):
-                    if st.session_state.button_clicked != "Who cares?":
-                        full_prompt = f"[INST]<<SYS>>\n{prompt + p_application}\n<</SYS>>\n\n{QUESTION}\n[/INST]"
-                        st.session_state.response = llm.invoke(full_prompt)
-                        st.session_state.retry_count = 0
-                        st.session_state.full_prompt = full_prompt
-                        st.session_state.button_clicked = "Who cares?"
+                        if st.session_state.button_clicked != "Who cares?":
+                            # First time clicked
+                            full_prompt = (
+                                "[INST]<<SYS>>\n"
+                                f"{prompt + p_application}\n"
+                                "<</SYS>>\n\n"
+                                f"{QUESTION}\n"
+                                "[/INST]"
+                            )
+                            st.session_state.response = llm.invoke(full_prompt)
+                            st.session_state.retry_count = 0
+                            st.session_state.full_prompt = full_prompt  # Save prompt for retry
+                            st.session_state.button_clicked = "Who cares?"
             st.markdown("---")
 
+        # Display response text if a sub-button is clicked
         if st.session_state.button_clicked:
             with st.container():
                 st.info(st.session_state.response)
-
-                # if st.session_state.button_clicked == "Explain the question":
-                #     col1, col2, col3 = st.columns([1, 1, 1])
-                #     with col2:
-                #         st.markdown(
-                #             """
-                #             <style>
-                #             div.stButton > button {
-                #                 width: 250px!important;
-                #             }
-                #             </style>
-                #             """,
-                #             unsafe_allow_html=True,
-                #         )
-                #         if st.button("I don't understand. Try again.", key="retry_button"):
-                #             st.session_state.retry_count += 1
-                #             alt_llm = get_llm(temperature=0.9)
-                #             st.session_state.response = alt_llm.invoke(st.session_state.full_prompt)
-                #             st.info(st.session_state.response)
-
-# Footer spacing
+        
+                if st.session_state.button_clicked == "Explain the question":
+                    if st.button("I don't understand. Try again."):
+                        st.session_state.retry_count += 1
+                        alt_llm = get_llm(temperature=0.9)
+                        st.session_state.response = alt_llm.invoke(
+                            st.session_state.full_prompt
+                        )
+# Optional: Add footer or spacing
 st.markdown("<br><br>", unsafe_allow_html=True)
 
-# Global button styling
+# css
 st.markdown(
     """
     <style>
@@ -165,4 +182,4 @@ st.markdown(
     </style>
     """,
     unsafe_allow_html=True,
-)
+)