Update app.py

app.py

@@ -81,5 +81,31 @@ def generate_summary_with_huggingface(query, retrieved_text):
         return "Error generating summary."
 
 def user_input(user_question, vector_store):
-    if vector_store i
+    if vector_store is None:
+        return "Vector store is empty."
+    try:
+        docs = vector_store.similarity_search(user_question)
+        context_text = " ".join([doc.page_content for doc in docs])
+        return generate_summary_with_huggingface(user_question, context_text)
+    except Exception as e:
+        st.error(f"Error in similarity search: {e}")
+        return "Error in similarity search."
+
+def main():
+    st.title("📄 Gen AI Lawyers Guide")
+    raw_text = fetch_pdf_text_from_folders(PDF_FOLDERS)
+    text_chunks = get_text_chunks(raw_text)
+    vector_store = load_or_create_vector_store(text_chunks)
+
+    user_question = st.text_input("Ask a Question:", placeholder="Type your question here...")
+
+    if st.button("Get Response"):
+        if not user_question:
+            st.warning("Please enter a question before submitting.")
+        else:
+            with st.spinner("Generating response..."):
+                answer = user_input(user_question, vector_store)
+                st.markdown(f"**🤖 AI:** {answer}")
 
+if __name__ == "__main__":
+    main()