Update app.py

app.py

@@ -64,34 +64,42 @@ def extract_text_from_pdfs(files):
     return text
 
 # Handle the query submission
-if submit_button and query:
-    # Extract text from uploaded PDFs
-    if uploaded_files:
-        pdf_text = extract_text_from_pdfs(uploaded_files)
-        
-        # Prepare the input prompt
-        prompt = f"""
-        Based on the following context/document:
-        {pdf_text}
-        Please answer the question: {query}
-        """
-
-        # Encode the input text
-        input_ids = tokenizer(prompt, return_tensors="pt").input_ids.to(device)
-
-        # Generate the response
-        outputs = model.generate(
-            input_ids=input_ids,
-            max_new_tokens=500,
-            no_repeat_ngram_size=5,
-        )
-
-        # Decode the response and clean it
-        response = tokenizer.decode(outputs[0], skip_special_tokens=True)
-        clean_response = response.strip()
-
-        # Update chat history
-        st.session_state.chat_history.append((query, clean_response))
+if submit_button:
+    if not query:
+        st.warning("⚠️ Please enter a query before submitting.")
+    elif not uploaded_files:
+        st.warning("⚠️ Please upload at least one PDF file before submitting.")
+    else:
+        try:
+            # Extract text from uploaded PDFs
+            pdf_text = extract_text_from_pdfs(uploaded_files)
+            
+            # Prepare the input prompt
+            prompt = f"""
+            Based on the following context/document:
+            {pdf_text}
+            Please answer the question: {query}
+            """
+
+            # Encode the input text
+            input_ids = tokenizer(prompt, return_tensors="pt").input_ids.to(device)
+
+            # Generate the response
+            outputs = model.generate(
+                input_ids=input_ids,
+                max_new_tokens=500,
+                no_repeat_ngram_size=5,
+            )
+
+            # Decode the response and clean it
+            response = tokenizer.decode(outputs[0], skip_special_tokens=True)
+            clean_response = response.strip()
+
+            # Update chat history
+            st.session_state.chat_history.append((query, clean_response))
+
+        except Exception as e:
+            st.error(f"An error occurred: {e}")
 
 # Display chat history
 if st.session_state.chat_history: