Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,5 +1,4 @@
 import os
-os.environ["STREAMLIT_HOME"] = "/tmp/.streamlit"
 from langchain_huggingface import HuggingFaceEndpoint
 import streamlit as st
 
@@ -19,24 +18,23 @@ 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"] = os.getenv("HF_TOKEN")  # Optional but ensures it's set
-
+    os.environ["HF_TOKEN"] = hf_token  # ensure token is accessible to endpoint
     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.
@@ -56,11 +54,10 @@ 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
@@ -75,9 +72,8 @@ 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="",
@@ -87,103 +83,74 @@ with st.container():
 
     st.markdown("")
 
-    # Help Button Logic
     def toggle_help():
         st.session_state.help_clicks += 1
-        st.session_state.button_clicked = None  # Reset help text on new toggle
+        st.session_state.button_clicked = None
 
-    # 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("---")  # Divider for clarity
+            st.markdown("---")
             col1, col2, col3 = st.columns(3)
 
             with col1:
                 if st.button("📝 Explain the question"):
                     if st.session_state.button_clicked != "Explain the question":
-                        # First time clicked
-                        full_prompt = (
-                            "[INST]<<SYS>>\n"
-                            f"{prompt + p_explanation}\n"
-                            "<</SYS>>\n\n"
-                            f"{QUESTION}\n"
-                            "[/INST]"
-                        )
+                        full_prompt = f"[INST]<<SYS>>\n{prompt + p_explanation}\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  # Save prompt for retry
+                        st.session_state.full_prompt = full_prompt
                         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":
-                            # 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"
+                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"
             with col3:
                 if st.button("🤔 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?"
+                    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?"
             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:
-                #         # Increase the width of the button to prevent text crowding
-                #         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
-                #             )
-                #             # Update the response text directly instead of rerunning the app
-                #             st.info(st.session_state.response)
-                                    
-# Optional: Add footer or spacing
+
+                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
 st.markdown("<br><br>", unsafe_allow_html=True)
 
-# css
+# Global button styling
 st.markdown(
     """
     <style>
@@ -199,4 +166,3 @@ st.markdown(
     """,
     unsafe_allow_html=True,
 )
-