Update app.py

app.py

@@ -71,6 +71,12 @@ def process_single_pdf(file_path):
 
 # Load PDFs with progress display
 def load_pdfs_with_progress(folder_path):
+    if not os.path.exists(folder_path):
+        st.error(f"The folder '{folder_path}' does not exist. Please create it and add PDF files.")
+        st.session_state['vector_store'] = None
+        st.session_state['loading'] = False
+        return
+
     all_text = ""
     pdf_files = [os.path.join(folder_path, filename) for filename in os.listdir(folder_path) if filename.endswith('.pdf')]
     num_files = len(pdf_files)
@@ -120,49 +126,4 @@ def user_input(user_question):
     vector_store = st.session_state.get('vector_store')
     if vector_store is None:
         return "The app is still loading documents or no documents were successfully loaded."
-    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)
-
-# Translate text to selected language
-def translate_text(text, target_lang):
-    translation_tokenizer.tgt_lang = target_lang
-    encoded_text = translation_tokenizer(text, return_tensors="pt")
-    generated_tokens = translation_model.generate(**encoded_text)
-    translated_text = translation_tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
-    return translated_text
-
-# Main function to run the Streamlit app
-def main():
-    st.markdown(
-        """
-        <h1 style="font-size:30px; text-align: center;">
-        📄 JusticeCompass: Your AI-Powered Legal Navigator for Swift, Accurate Guidance.
-        </h1>
-        """,
-        unsafe_allow_html=True
-    )
-
-    if 'loading' not in st.session_state or st.session_state['loading']:
-        st.session_state['loading'] = True
-        load_pdfs_with_progress('documents1')
-
-    user_question = st.text_input("Ask a Question:", placeholder="Type your question here...")
-
-    # Display language selection dropdown
-    selected_language = st.selectbox("Select output language:", list(LANGUAGES.keys()))
-
-    if st.session_state.get('loading', True):
-        st.info("The app is loading documents in the background. You can type your question now and submit once loading is complete.")
-
-    # Only display "Get Response" button after user enters a question
-    if user_question:
-        if st.button("Get Response"):
-            with st.spinner("Generating response..."):
-                answer = user_input(user_question)
-                target_lang_code = LANGUAGES[selected_language]
-                translated_answer = translate_text(answer, target_lang_code)
-                st.markdown(f"**🤖 AI ({selected_language}):** {translated_answer}")
-
-if __name__ == "__main__":
-    main()
+    docs = vector_store.simi