Update app.py

app.py

@@ -58,10 +58,8 @@ def validate_prompt(prompt: str) -> bool:
     Returns:
         bool: True if the prompt is valid, False otherwise.
     """
-    # Improved validation: Allow alphanumeric characters, spaces, and punctuation
-    if re.match(r'^[A-Za-z0-9\s\.,;!?(){}[\]]+$', prompt):
-        return True
-    return False
+    # Allow alphanumeric characters, spaces, and punctuation
+    return bool(re.match(r'^[A-Za-z0-9\s\.,;!?(){}[\]]+$', prompt))
 
 # Function to convert text to speech
 def text_to_speech(text: str) -> None:
@@ -73,15 +71,10 @@ def text_to_speech(text: str) -> None:
     try:
         tts = gTTS(text, lang='en')
         tts.save("response.mp3")
-        os.system("mpg321 response.mp3")
+        os.system("mpg321 response.mp3")  # Use mpg321 to play the audio
     except Exception as e:
         st.error(f"Error generating speech: {e}")
 
-# Display chat history
-for message in st.session_state.messages:
-    with st.chat_message(message["role"]):
-        st.write(message["content"])
-
 # Function to generate chatbot response
 def generate_response(prompt: str) -> str:
     """
@@ -112,29 +105,35 @@ def generate_response(prompt: str) -> str:
 
     return response
 
+# Display chat history using native Streamlit components
+for message in st.session_state.messages:
+    if message["role"] == "user":
+        st.markdown(f"**You:** {message['content']}")
+    elif message["role"] == "assistant":
+        st.markdown(f"**Assistant:** {message['content']}")
+
 # User input handling
-if prompt := st.chat_input():
+prompt = st.text_input("Your message:", key="chat_input")
+if prompt:
     # Validate user input
     if validate_prompt(prompt):
         st.session_state.messages.append({"role": "user", "content": prompt})
-        with st.chat_message("user"):
-            st.write(prompt)
+        st.markdown(f"**You:** {prompt}")
 
-        # Generate and display response with smooth animations
-        with st.chat_message("assistant"):
-            with st.spinner("Assistant is typing..."):
-                response = generate_response(prompt)
-                st.write(response)
-        
-        # Text-to-Speech integration for the assistant's response
-        text_to_speech(response)
+        # Generate assistant response
+        with st.spinner("Assistant is typing..."):
+            response = generate_response(prompt)
+            st.markdown(f"**Assistant:** {response}")
 
         # Store the assistant's response
         st.session_state.messages.append({"role": "assistant", "content": response})
+
+        # Text-to-Speech integration for the assistant's response
+        text_to_speech(response)
     else:
         st.warning("Invalid input. Please ensure your input contains only valid characters.")
 
 # User Feedback Section
 feedback = st.selectbox("How was your experience?", ["😊 Excellent", "😐 Okay", "😕 Poor"])
 if feedback:
-    st.success(f"Thank you for your feedback: {feedback}", icon="✅")
+    st.success(f"Thank you for your feedback: {feedback}", icon="✅")