Update app.py

app.py

@@ -1,4 +1,11 @@
 import streamlit as st
+
+# IMPORTANT: Call set_page_config before any other Streamlit command
+st.set_page_config(
+    page_title="TARS: Therapist Assistance and Response System", 
+    page_icon="🧠"
+)
+
 from transformers import pipeline
 import torch
 
@@ -28,19 +35,20 @@ class AITherapistAssistant:
                 device=0 if torch.cuda.is_available() else -1
             )
             
-            # Summarization model (using a validated model)
+            # Summarization model
             self.summary_model = pipeline(
                 "summarization", 
                 model="facebook/bart-large-cnn", 
                 device=0 if torch.cuda.is_available() else -1
             )
         except Exception as e:
+            # Display error if model loading fails
             st.error(f"Model loading error: {e}")
             self.conversation_model = None
             self.summary_model = None
     
     def detect_crisis(self, message):
-        """Detect potential suicide risk in message"""
+        """Detect potential suicide risk in message."""
         message_lower = message.lower()
         for keyword in SUICIDE_KEYWORDS:
             if keyword in message_lower:
@@ -48,14 +56,16 @@ class AITherapistAssistant:
         return False
     
     def generate_response(self, message):
-        """Generate empathetic AI response"""
+        """Generate empathetic AI response."""
         if not self.conversation_model:
             return "I'm here to listen. Would you like to share more about how you're feeling?"
         
         try:
             # Contextual prompt to guide response
-            full_prompt = f"You are a compassionate AI therapist. Respond supportively to this message: {message}. Be empathetic, validate feelings, and avoid giving direct medical advice."
-            
+            full_prompt = (
+                "You are a compassionate AI therapist. Respond supportively to this message: "
+                f"{message}. Be empathetic, validate feelings, and avoid giving direct medical advice."
+            )
             # Generate response
             response = self.conversation_model(
                 full_prompt, 
@@ -65,37 +75,33 @@ class AITherapistAssistant:
             
             return response
         
-        except Exception as e:
+        except Exception:
             return "I'm here to listen. Would you like to share more about how you're feeling?"
     
     def generate_summary(self, conversation):
-        """Generate a professional therapy-style summary"""
+        """Generate a professional therapy-style summary."""
         if not self.summary_model:
             return "Summary generation is temporarily unavailable."
         
         try:
-            # Generate summary
             summary = self.summary_model(
                 conversation, 
                 max_length=130, 
                 min_length=30, 
                 do_sample=False
             )[0]['summary_text']
-            
             return summary
         
-        except Exception as e:
+        except Exception:
             return "Summary could not be generated."
 
 def main():
-    st.set_page_config(
-        page_title="TARS: Therapist Assistance and Response System", 
-        page_icon="🧠"
-    )
-    
+    # Title and description
     st.title("🧠 TARS: Therapist Assistance and Response System")
-    st.write("A supportive space to share your feelings safely")
-    
+    st.write("A supportive space to share your feelings safely.\n\n"
+             "**Disclaimer**: I am not a licensed therapist. "
+             "If you're in crisis, please reach out to professional help immediately.")
+
     # Initialize session state
     if 'conversation' not in st.session_state:
         st.session_state.conversation = []
@@ -103,14 +109,14 @@ def main():
     if 'assistant' not in st.session_state:
         st.session_state.assistant = AITherapistAssistant()
     
-    # Conversation Display
+    # Display conversation
     for message in st.session_state.conversation:
         if message['sender'] == 'user':
             st.chat_message("user").write(message['text'])
         else:
             st.chat_message("assistant").write(message['text'])
     
-    # User Input
+    # User input with chat_input
     if prompt := st.chat_input("Share your thoughts. I'm here to listen."):
         # Check for crisis indicators
         if st.session_state.assistant.detect_crisis(prompt):
@@ -119,13 +125,8 @@ def main():
             for org, phone in CRISIS_RESOURCES.items():
                 st.markdown(f"- {org}: `{phone}`")
         
-        # Add user message
-        st.session_state.conversation.append({
-            'sender': 'user', 
-            'text': prompt
-        })
-        
-        # Display user message
+        # Add user message to conversation
+        st.session_state.conversation.append({'sender': 'user', 'text': prompt})
         st.chat_message("user").write(prompt)
         
         # Generate AI response
@@ -134,16 +135,13 @@ def main():
                 ai_response = st.session_state.assistant.generate_response(prompt)
                 st.write(ai_response)
         
-        # Add AI message
-        st.session_state.conversation.append({
-            'sender': 'ai', 
-            'text': ai_response
-        })
+        # Add AI response to conversation
+        st.session_state.conversation.append({'sender': 'ai', 'text': ai_response})
     
     # Session Summary Generation
     if st.session_state.conversation:
         if st.button("Generate Session Summary"):
-            conversation_text = " ".join([msg['text'] for msg in st.session_state.conversation])
+            conversation_text = " ".join(msg['text'] for msg in st.session_state.conversation)
             summary = st.session_state.assistant.generate_summary(conversation_text)
             st.markdown("**Session Summary:**")
             st.write(summary)
@@ -156,9 +154,3 @@ def main():
 
 if __name__ == "__main__":
     main()
-
-# requirements.txt
-# streamlit
-# transformers
-# torch
-# accelerate