Update app.py

app.py

@@ -1,101 +1,80 @@
 import streamlit as st
 import time
 
-# FIRST: Set page config before ANY other Streamlit command
 st.set_page_config(page_title="Indian Spiritual RAG")
 
-# Initialize ALL session state variables
+# Initialize session state variables
 if 'initialized' not in st.session_state:
     st.session_state.initialized = False
-if 'model' not in st.session_state:
-    st.session_state.model = None
-if 'tokenizer' not in st.session_state:
-    st.session_state.tokenizer = None
 if 'last_query' not in st.session_state:
     st.session_state.last_query = ""
 if 'submit_clicked' not in st.session_state:
     st.session_state.submit_clicked = False
-if 'query_answered' not in st.session_state:
-    st.session_state.query_answered = False  # Tracks if query was answered
+if 'query_input' not in st.session_state:
+    st.session_state.query_input = ""
 
-# THEN: Import your modules
+# Import other modules
 from rag_engine import process_query, load_model
 from utils import setup_all_auth
 
-# Create a placeholder for initialization message
-init_message = st.empty()
-
-# ✅ Custom CSS for border & button fixes
+# Custom CSS for styling
 st.markdown("""
 <style>
-/* ✅ Green border by default */
-div[data-baseweb="input"] {
-    border: 2px solid #4CAF50 !important;
-    border-radius: 8px !important;
-    transition: border 0.3s ease-in-out;
-}
-
-/* 🔴 Change to red while typing */
-div[data-baseweb="input"]:focus-within {
-    border: 2px solid #FF5722 !important;
+.main-title {
+    font-size: 2.5rem;
+    color: #FF5722;
+    text-align: center;
+    margin-bottom: 1rem;
 }
-
-/* ✅ Ensure it turns green again AFTER submission */
-.answered div[data-baseweb="input"] {
-    border: 2px solid #4CAF50 !important;
-}
-
-/* ✅ Fully green button */
 .stButton>button {
     background-color: #4CAF50 !important;
     color: white !important;
-    border: 2px solid #4CAF50 !important;
     border-radius: 8px !important;
-    padding: 8px 16px;
-    font-size: 16px;
-    font-weight: bold;
-    cursor: pointer;
+    font-weight: bold !important;
+}
+div[data-baseweb="input"] {
+    border: 3px solid #4CAF50 !important;
+    border-radius: 8px !important;
+    transition: border-color 0.3s ease-in-out;
+}
+div[data-baseweb="input"]:focus-within {
+    border: 3px solid #FF5722 !important;
 }
 </style>
+<div class="main-title">Indian Spiritual Texts Q&A</div>
 """, unsafe_allow_html=True)
 
-# Handle initialization
+# Initialization logic
 if not st.session_state.initialized:
+    init_message = st.empty()
     init_message.info("Hang in there! We are setting the system up for you. 😊")
-
     try:
         setup_all_auth()
         load_model()
         st.session_state.initialized = True
-        st.session_state.query_answered = False  # Reset query state
+        time.sleep(0.5)
         init_message.success("System initialized successfully!")
         time.sleep(1)
         init_message.empty()
     except Exception as e:
         init_message.error(f"Error initializing: {str(e)}")
 
-# ✅ Create a form for user input
+# Function to handle form submission
+def handle_form_submit():
+    st.session_state.last_query = st.session_state.query_input
+    st.session_state.submit_clicked = True
+    st.session_state.query_input = ""  # ✅ Clears input field on submission
+
+# Form for user input
 with st.form(key="query_form"):
     query = st.text_input(
-        "Ask your question:",
-        key="query_input",
-        placeholder="Type here...",
+        "Ask your question:", 
+        key="query_input", 
+        value=st.session_state.query_input
     )
-    submit_button = st.form_submit_button("Get Answer")
-
-# ✅ Handle submission **without st.rerun()**
-if submit_button:
-    if query.strip():
-        st.session_state.last_query = query.strip()
-        st.session_state.submit_clicked = True
-        st.session_state.query_answered = True  # ✅ Ensure text box resets to green
-
-# ✅ Wrap the text input inside a div for styling
-css_class = "answered" if st.session_state.query_answered else ""
-st.markdown(f'<div class="{css_class}">', unsafe_allow_html=True)
-st.markdown("</div>", unsafe_allow_html=True)  # ✅ Close styling div
+    submit_button = st.form_submit_button("Get Answer", on_click=handle_form_submit)
 
-# ✅ Display the last question after submission
+# Display last question
 if st.session_state.last_query:
     st.markdown("### Current Question:")
     st.info(st.session_state.last_query)
@@ -107,25 +86,21 @@ with col1:
 with col2:
     word_limit = st.slider("Word limit:", 50, 500, 200)
 
-# ✅ Process the query if submitted
-if st.session_state.submit_clicked:
-    st.session_state.submit_clicked = False  # Reset flag
-
+# Process query when submitted
+if st.session_state.submit_clicked and st.session_state.last_query:
+    st.session_state.submit_clicked = False
     with st.spinner("Processing your question..."):
         try:
             result = process_query(st.session_state.last_query, top_k=top_k, word_limit=word_limit)
-
             st.subheader("Answer:")
             st.write(result["answer_with_rag"])
-
             st.subheader("Sources:")
             for citation in result["citations"].split("\n"):
                 st.write(citation)
-
         except Exception as e:
             st.error(f"Error processing query: {str(e)}")
 
-# Add helpful information
+# Footer
 st.markdown("---")
 st.markdown("""
 ### About this app