Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,19 +1,16 @@
-import os, pathlib
+import os
+import pathlib
 
-
-
-os.environ["HOME"] = "/tmp"       
+os.environ["HOME"] = "/tmp"
 cfg_dir = pathlib.Path("/tmp/.streamlit")
 cfg_dir.mkdir(parents=True, exist_ok=True)
-
 (cfg_dir / "config.toml").write_text(
-    "[server]\nheadless = true\nport = 7860\naddress = \"0.0.0.0\"\n\n"
-    "[browser]\ngatherUsageStats = false\n",
+    "[server]\nheadless = true\nport = 7860\naddress = \"0.0.0.0\"\n\n[browser]\ngatherUsageStats = false\n",
     encoding="utf-8",
 )
-import streamlit as st
 
 import time
+import streamlit as st
 from huggingface_hub import InferenceClient
 import re
 import gspread
@@ -27,18 +24,15 @@ from huggingface_hub.utils import HfHubHTTPError
 import httpx
 
 
-# Configure logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# Set Streamlit page config early
 st.set_page_config(
     page_title="Interactive Help Interface", 
     layout="centered",
     initial_sidebar_state="collapsed"
 )
 
-# Constants
 DEFAULT_QUESTIONS = [
     "Please familiarize yourself with all the buttons before completing the questions below.",
     "What is sin(π/6)?",
@@ -47,7 +41,6 @@ DEFAULT_QUESTIONS = [
     "Find the vertex (x,y) for a parabola with equation:\ny=3x^2 -6x + 1",
     "What are the solutions to y = x^2 + x - 6?"
 ]
-# text generation 
 MODEL = "deepseek-ai/DeepSeek-R1-0528"
 SUBJECT = "Precalculus"
 WORD_LIMIT = 180
@@ -57,9 +50,6 @@ class ConfigManager:
         self.hf_token = os.getenv("HF_TOKEN")
         self.google_creds_json = os.getenv("GOOGLE_SHEETS_CREDENTIALS")
         self.google_sheets_id = os.getenv("GOOGLE_SHEETS_ID")
-        
-
-
         missing_vars = []
 
     def get_google_client(self):
@@ -103,7 +93,6 @@ class GoogleSheetsLogger:
             spreadsheet = client.open_by_key(sheet_id)
             sheet = spreadsheet.sheet1
 
-            # Initialize headers if not present
             headers = sheet.row_values(1)
             if not headers:
                 sheet.append_row([
@@ -115,7 +104,6 @@ class GoogleSheetsLogger:
 
         except Exception as e:
             logger.error(f"Error connecting to Google Sheets: {e}")
-            # Do NOT crash the UI; just return None so logging becomes a no-op
             return None
 
     def log_button_click(self, button_name, question, subject, help_clicks, retry_count):
@@ -141,17 +129,11 @@ class GoogleSheetsLogger:
 
 
 class AIAssistant:
-    def __init__(self, config):
-        self.model =  os.getenv("HF_MODEL",  "HuggingFaceH4/zephyr-7b-beta") 
-        token = os.getenv("HF_TOKEN")
-        if os.getenv("SPACE_ID") or os.getenv("HF_SPACE_ID"):
-            token =  None 
-
-        self.client = InferenceClient(
-            model=self.model,
-            token=token,
-            timeout=60.0,
-        )
+    """Handles AI model interactions"""
+    
+    def __init__(self, config: ConfigManager):
+        self.model = MODEL
+        self.client = InferenceClient(model=self.model, token=config.hf_token, timeout=60.0)
 
         self.base_prompt = f"""
 You are an AI assistant designed to support high school students in the subject of {SUBJECT}. 
@@ -187,34 +169,46 @@ Please follow these guidelines:
                 system_text += f"\nPlease provide a different explanation. This is attempt {retry_count + 1}."
             user_text = f"Question:\n{question}"
 
+            full_prompt = f"{system_text}\n\n{user_text}"
+
             try:
-                messages = [
-                    {"role": "system", "content": system_text},
-                    {"role": "user", "content": user_text},
-                ]
-                chat = self.client.chat_completion(
-                    messages=messages,
-                    max_tokens=350,
-                    temperature=0.7,
-                )
-                text = ""
-                try:
-                    choices = getattr(chat, "choices", None) or chat.get("choices", [])
-                    if choices:
-                        msg0 = choices[0].get("message") or {}
-                        text = msg0.get("content") or ""
-                    if not text:
-                        text = getattr(chat, "generated_text", None) or chat.get("generated_text", "") or ""
-                except Exception:
-                    text = str(chat)
-            except (HfHubHTTPError, ValueError, AttributeError):
-                full_prompt = f"{system_text}\n\n{user_text}"
                 text = self.client.text_generation(
                     prompt=full_prompt,
                     max_new_tokens=300,
                     temperature=0.7,
                     repetition_penalty=1.1,
+                    model=self.model,
+                )
+            except (HfHubHTTPError, ValueError) as e:
+                msg = str(e)
+                unsupported = (
+                    "Task 'text-generation' not supported" in msg
+                    or "doesn't support task 'text-generation'" in msg
+                    or "Available tasks: ['conversational']" in msg
                 )
+                if unsupported:
+                    messages = [
+                        {"role": "system", "content": system_text},
+                        {"role": "user", "content": user_text},
+                    ]
+                    chat = self.client.chat_completion(
+                        messages=messages,
+                        max_tokens=350,
+                        temperature=0.7,
+                        model=self.model,
+                    )
+                    text = ""
+                    try:
+                        choices = getattr(chat, "choices", None) or chat.get("choices", [])
+                        if choices:
+                            msg0 = choices[0].get("message") or {}
+                            text = msg0.get("content") or ""
+                        if not text:
+                            text = getattr(chat, "generated_text", None) or chat.get("generated_text", "") or ""
+                    except Exception:
+                        text = str(chat)
+                else:
+                    raise
             except (httpx.ReadTimeout, httpx.ConnectTimeout):
                 return "The model request timed out. Please try again."
 
@@ -233,16 +227,15 @@ class SessionManager:
     
     @staticmethod
     def initialize_session_state():
-        """Initialize session state variables"""
         defaults = {
-            "responses": {},  # Store responses per question
-            "retry_counts": {},  # Store retry counts per question
-            "help_clicks": {},  # Store help clicks per question
-            "button_clicked": {},  # Store which button was clicked per question
-            "show_sections": {},  # Store section visibility per question
-            "help_dismissed": {},  # Store help dismissed state per question
-            "loading_states": {},  # Store loading states per question
-            "submitted": {},  # Store submission state per question
+            "responses": {},
+            "retry_counts": {},
+            "help_clicks": {},
+            "button_clicked": {},
+            "show_sections": {},
+            "help_dismissed": {},
+            "loading_states": {},
+            "submitted": {},
             "session_id": str(uuid.uuid4())
         }
         
@@ -252,14 +245,12 @@ class SessionManager:
     
     @staticmethod
     def get_question_state(question_id: str, key: str, default=None):
-        """Get state for a specific question"""
         if key not in st.session_state:
             st.session_state[key] = {}
         return st.session_state[key].get(question_id, default)
     
     @staticmethod
     def set_question_state(question_id: str, key: str, value):
-        """Set state for a specific question"""
         if key not in st.session_state:
             st.session_state[key] = {}
         st.session_state[key][question_id] = value
@@ -275,13 +266,9 @@ class HelpInterface:
         self.session_manager.initialize_session_state()
     
     def render_question(self, question: str, question_id: str, is_demo: bool = False):
-        """Render a single question interface"""
-        # Debug logging
         logger.info(f"Rendering question {question_id}, is_demo: {is_demo}")
         
-        # Question container with enhanced styling
         if is_demo:
-            # Demo question styling
             st.markdown(f"""
             <div style='
                 background: linear-gradient(135deg, #fdcb6e 0%, #e17055 100%);
@@ -314,7 +301,6 @@ class HelpInterface:
             </div>
             """, unsafe_allow_html=True)
         else:
-            # Regular question styling
             st.markdown(f"""
             <div style='
                 background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
@@ -347,10 +333,8 @@ class HelpInterface:
             </div>
             """, unsafe_allow_html=True)
         
-        # Check if question is submitted
         is_submitted = self.session_manager.get_question_state(question_id, "submitted", False)
         
-        # Student response area with improved spacing - only show if not submitted
         if not is_submitted:
             st.markdown("<div style='margin: 20px 0;'>", unsafe_allow_html=True)
             st.text_area(
@@ -362,7 +346,6 @@ class HelpInterface:
             )
             st.markdown("</div>", unsafe_allow_html=True)
         
-        # Enhanced button section with better spacing and styling
         st.markdown("""
         <div style='
             margin: 30px 0 20px 0;
@@ -371,7 +354,6 @@ class HelpInterface:
         '>
         """, unsafe_allow_html=True)
         
-        # Button layout with improved spacing
         col1, col2, col3 = st.columns([1, 0.2, 1])
         
         with col1:
@@ -404,11 +386,9 @@ class HelpInterface:
                 if st.button("🔍 Need Help?", key=f"help_{question_id}", type="primary", use_container_width=True):
                     self._toggle_help(question, question_id, is_demo)
             else:
-                # Empty space to maintain layout when submitted
                 st.markdown("<div style='height: 55px;'></div>", unsafe_allow_html=True)
         
         with col2:
-            # Empty column for spacing
             st.markdown("<div style='height: 55px;'></div>", unsafe_allow_html=True)
         
         with col3:
@@ -441,7 +421,6 @@ class HelpInterface:
                 if st.button("📝 Submit Answer", key=f"submit_{question_id}", use_container_width=True):
                     self._handle_submit(question, question_id, is_demo)
             else:
-                # Submitted state with enhanced styling
                 st.markdown("""
                 <div style='
                     background: linear-gradient(135deg, #48bb78 0%, #38a169 100%);
@@ -462,25 +441,19 @@ class HelpInterface:
                 </div>
                 """, unsafe_allow_html=True)
         
-        # Close button container
         st.markdown("</div>", unsafe_allow_html=True)
         
-        # Render help section if active and not submitted
         if (self.session_manager.get_question_state(question_id, "show_sections", False) and 
             not is_submitted):
             self._render_help_section(question, question_id, is_demo)
         
-        # Add spacing between questions
         st.markdown("<div style='margin-bottom: 40px;'></div>", unsafe_allow_html=True)
     
     def _handle_submit(self, question: str, question_id: str, is_demo: bool = False):
-        """Handle submit button click"""
         logger.info(f"Handle submit called for question {question_id}, is_demo: {is_demo}")
         
-        # Set submission state
         self.session_manager.set_question_state(question_id, "submitted", True)
         
-        # Log submit only if not demo
         if not is_demo:
             help_clicks = self.session_manager.get_question_state(question_id, "help_clicks", 0)
             retry_count = self.session_manager.get_question_state(question_id, "retry_counts", 0)
@@ -495,7 +468,6 @@ class HelpInterface:
         st.rerun()
     
     def _toggle_help(self, question: str, question_id: str, is_demo: bool = False):
-        """Toggle help section visibility"""
         logger.info(f"Toggle help called for question {question_id}, is_demo: {is_demo}")
         
         current_help_clicks = self.session_manager.get_question_state(question_id, "help_clicks", 0)
@@ -504,11 +476,9 @@ class HelpInterface:
         current_show = self.session_manager.get_question_state(question_id, "show_sections", False)
         self.session_manager.set_question_state(question_id, "show_sections", not current_show)
         
-        # Reset other states
         self.session_manager.set_question_state(question_id, "button_clicked", None)
         self.session_manager.set_question_state(question_id, "help_dismissed", False)
         
-        # Log the help button click only if not demo
         if not is_demo:
             retry_count = self.session_manager.get_question_state(question_id, "retry_counts", 0)
             logger.info(f"Logging help click for question {question_id}")
@@ -520,7 +490,6 @@ class HelpInterface:
             logger.info(f"Skipping logging for demo question {question_id}")
     
     def _render_help_section(self, question: str, question_id: str, is_demo: bool = False):
-        """Render the help section with options"""
         st.markdown("""
         <div style='
             background: linear-gradient(135deg, #ffeaa7 0%, #fab1a0 100%);
@@ -543,7 +512,6 @@ class HelpInterface:
         </div>
         """, unsafe_allow_html=True)
         
-        # Help option buttons in a more compact layout
         col1, col2, col3 = st.columns(3)
         
         with col1:
@@ -558,21 +526,16 @@ class HelpInterface:
             if st.button("🤔 Who cares?", key=f"application_{question_id}", use_container_width=True):
                 self._handle_help_button("Who cares?", question, question_id, is_demo)
         
-        # Show response if available - NO GREEN BAR
         current_response = self.session_manager.get_question_state(question_id, "responses", "")
         current_button = self.session_manager.get_question_state(question_id, "button_clicked", None)
         
         if current_response and current_button:
-            # Simply display the response without any container styling
             st.write(current_response)
-            
             self._render_action_buttons(question, question_id, is_demo)
     
     def _handle_help_button(self, button_name: str, question: str, question_id: str, is_demo: bool = False):
-        """Handle help button click and generate response"""
         logger.info(f"Handle help button called: {button_name} for question {question_id}, is_demo: {is_demo}")
         
-        # Log button click only if not demo
         if not is_demo:
             help_clicks = self.session_manager.get_question_state(question_id, "help_clicks", 0)
             retry_count = self.session_manager.get_question_state(question_id, "retry_counts", 0)
@@ -584,12 +547,9 @@ class HelpInterface:
         else:
             logger.info(f"Skipping logging for demo question {question_id}")
         
-        # Generate response
         with st.spinner("Generating response..."):
             retry_count = self.session_manager.get_question_state(question_id, "retry_counts", 0)
             response = self.ai_assistant.generate_response(button_name, question, retry_count)
-            
-            # Store response and button clicked
             self.session_manager.set_question_state(question_id, "responses", response)
             self.session_manager.set_question_state(question_id, "button_clicked", button_name)
             self.session_manager.set_question_state(question_id, "retry_counts", 0)
@@ -597,7 +557,6 @@ class HelpInterface:
         st.rerun()
     
     def _render_action_buttons(self, question: str, question_id: str, is_demo: bool = False):
-        """Render action buttons after showing response"""
         st.markdown("<div style='margin-top: 15px;'>", unsafe_allow_html=True)
         col1, col2, col3 = st.columns(3)
         
@@ -616,7 +575,6 @@ class HelpInterface:
         st.markdown("</div>", unsafe_allow_html=True)
     
     def _handle_retry(self, question: str, question_id: str, is_demo: bool = False):
-        """Handle retry button click"""
         logger.info(f"Handle retry called for question {question_id}, is_demo: {is_demo}")
         
         current_retry = self.session_manager.get_question_state(question_id, "retry_counts", 0)
@@ -624,7 +582,6 @@ class HelpInterface:
         
         self.session_manager.set_question_state(question_id, "retry_counts", new_retry_count)
         
-        # Log retry only if not demo
         if not is_demo:
             help_clicks = self.session_manager.get_question_state(question_id, "help_clicks", 0)
             logger.info(f"Logging retry for question {question_id}")
@@ -634,7 +591,6 @@ class HelpInterface:
         else:
             logger.info(f"Skipping retry logging for demo question {question_id}")
         
-        # Generate new response
         current_button = self.session_manager.get_question_state(question_id, "button_clicked", None)
         if current_button:
             with st.spinner("Generating alternative response..."):
@@ -643,10 +599,8 @@ class HelpInterface:
             st.rerun()
     
     def _handle_chatgpt_redirect(self, question: str, question_id: str, is_demo: bool = False):
-        """Handle ChatGPT redirect"""
         logger.info(f"Handle ChatGPT redirect called for question {question_id}, is_demo: {is_demo}")
         
-        # Log only if not demo
         if not is_demo:
             help_clicks = self.session_manager.get_question_state(question_id, "help_clicks", 0)
             retry_count = self.session_manager.get_question_state(question_id, "retry_counts", 0)
@@ -675,10 +629,8 @@ class HelpInterface:
         """, unsafe_allow_html=True)
     
     def _handle_thank_you(self, question: str, question_id: str, is_demo: bool = False):
-        """Handle thank you button click"""
         logger.info(f"Handle thank you called for question {question_id}, is_demo: {is_demo}")
         
-        # Log only if not demo
         if not is_demo:
             help_clicks = self.session_manager.get_question_state(question_id, "help_clicks", 0)
             retry_count = self.session_manager.get_question_state(question_id, "retry_counts", 0)
@@ -690,7 +642,6 @@ class HelpInterface:
         else:
             logger.info(f"Skipping thank you logging for demo question {question_id}")
         
-        # Hide help section
         self.session_manager.set_question_state(question_id, "show_sections", False)
         self.session_manager.set_question_state(question_id, "help_dismissed", True)
         
@@ -728,7 +679,6 @@ def _get_sheet(_creds_json: str, _sheet_id: str):
 
 
 def main():
-    """Main application entry point"""
     st.markdown("""
     <div style='
         text-align: center;
@@ -753,25 +703,19 @@ def main():
     </div>
     """.format(SUBJECT=SUBJECT), unsafe_allow_html=True)
     
-    # Initialize configuration and interface
     config = ConfigManager()
     interface = HelpInterface(config)
     
-    # Render questions with proper numbering
     question_number = 1
     for i, question in enumerate(DEFAULT_QUESTIONS):
         if i == 0:
-            # First question is demo - no logging, special styling
             interface.render_question(question, "demo", is_demo=True)
         else:
-            # Regular questions with logging and proper numbering (1, 2, 3, 4, 5)
             interface.render_question(question, str(question_number), is_demo=False)
             question_number += 1
     
-    # Add custom CSS for enhanced styling
     st.markdown("""
     <style>
-    /* Enhanced button styling */
     .stButton > button {
         height: 2.5em;
         font-size: 0.9rem;
@@ -786,13 +730,11 @@ def main():
         box-shadow: 0 6px 20px rgba(0,0,0,0.15);
     }
     
-    /* Primary button styling */
     .stButton > button[kind="primary"] {
         background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
         color: white;
     }
     
-    /* Text area styling */
     .stTextArea > div > div > textarea {
         border-radius: 10px;
         border: 2px solid #e0e4e7;
@@ -812,18 +754,15 @@ def main():
         box-shadow: 0 0 0 3px rgba(102, 126, 234, 0.1);
     }
     
-    /* Remove default spacing */
     .block-container {
         padding-top: 2rem;
         padding-bottom: 2rem;
     }
     
-    /* Custom spacing */
     .stMarkdown {
         margin-bottom: 0;
     }
     
-    /* Success message styling */
     .stSuccess {
         background: linear-gradient(135deg, #00b894 0%, #00a085 100%);
         border: none;
@@ -832,7 +771,6 @@ def main():
         color: white;
     }
     
-    /* Spinner styling */
     .stSpinner {
         color: #667eea;
     }