Update app.py

app.py

@@ -4,10 +4,9 @@ import torch
 from PIL import Image
 import base64
 
-
 st.set_page_config(page_title="Menstrual Health Chatbot 💬", layout="centered")
 
-
+# Load model
 @st.cache_resource
 def load_model():
     tokenizer = AutoTokenizer.from_pretrained("adi2606/Menstrual-Health-Awareness-Chatbot")
@@ -16,14 +15,14 @@ def load_model():
 
 tokenizer, model = load_model()
 
-
+# Generate response
 def generate_response(input_text):
     inputs = tokenizer(input_text, return_tensors="pt")
     outputs = model.generate(**inputs, max_length=128)
     response = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return response
 
-
+# Background image
 def set_background(image_path):
     with open(image_path, "rb") as f:
         encoded_string = base64.b64encode(f.read()).decode()
@@ -40,21 +39,28 @@ def set_background(image_path):
         )
 
 
-set_background("background.jpg") 
+set_background("background.jpg")  
 
 
-banner = Image.open("banner.png")  
-st.image(banner, use_column_width=True)
+banner = Image.open("banner.png")
+st.image(banner, width=100)  
 
 
+st.markdown("""
+    <style>
+    h1, h4, .stMarkdown p, .stTextInput label, .css-1d391kg {
+        color: #111111 !important;
+    }
+    </style>
+""", unsafe_allow_html=True)
+
+# App content
 st.markdown("<h1 style='text-align: center;'>🩸 Menstrual Health Awareness Chatbot 💬</h1>", unsafe_allow_html=True)
 st.markdown("<h4 style='text-align: center;'>Ask anything about periods, PMS, hygiene, and more!</h4>", unsafe_allow_html=True)
 
-
 st.markdown("### 🤔 Your Question")
 user_input = st.text_input("", placeholder="E.g., What are the symptoms of PMS?")
 
-
 if st.button("🚀 Ask"):
     if user_input.strip():
         with st.spinner("Generating a helpful response..."):
@@ -64,6 +70,5 @@ if st.button("🚀 Ask"):
     else:
         st.warning("⚠️ Please enter a question to get started.")
 
-
 st.markdown("---")
 st.markdown("<small style='color: gray;'>Made with ❤️ to spread awareness.</small>", unsafe_allow_html=True)