Update app.py

app.py

@@ -67,8 +67,8 @@ def generate_response(query, context):
     return tokenizer.batch_decode(output, skip_special_tokens=True)[0]
 
 # Streamlit UI
-st.set_page_config(page_title="📖 RAG-Based AI", page_icon="🤖")
-st.title("📖 RAG-based Q&A using IBM Granite")
+st.set_page_config(page_title="📖 Smart Study", page_icon="🤖")
+st.title("📖 Q&A using IBM Granite")
 st.subheader("Upload documents and ask questions!")
 
 uploaded_files = st.file_uploader("Upload PDFs or TXT files", accept_multiple_files=True)
@@ -78,10 +78,13 @@ if uploaded_files:
         process_documents(uploaded_files)
     st.success("Documents uploaded and indexed!")
 
-query = st.text_input("Ask a question:")
-if st.button("Get Answer"):
-    with st.spinner("Retrieving and generating response..."):
-        context = retrieve_context(query)
-        response = generate_response(query, context)
-        st.markdown("### 🤖 Answer:")
-        st.write(response)
+if query:
+    if index.ntotal == 0:  # Ensure documents are indexed before querying
+        st.warning("Please upload and process documents first!")
+    else:
+        with st.spinner("Retrieving and generating response..."):
+            context = retrieve_context(query)
+            response = generate_response(query, context)
+            st.markdown("### 🤖 Answer:")
+            st.write(response)
+