Update app.py

app.py

@@ -1,4 +1,7 @@
 import os
+# --- FIX 1: Set User Agent to prevent WebBaseLoader crash ---
+os.environ["USER_AGENT"] = "Mozilla/5.0 (Windows NT 10.0; Win64; x64)"
+
 import gradio as gr
 from langchain_community.document_loaders import WebBaseLoader, PyMuPDFLoader
 from langchain_huggingface import HuggingFaceEmbeddings
@@ -6,14 +9,9 @@ from langchain_community.vectorstores import FAISS
 from langchain_community.llms import HuggingFaceHub
 from langchain.chains.question_answering import load_qa_chain
 
-# --- CONFIGURATION ---
-# We get the token from the Space's secret environment variables
+# Get the token from the secrets
 hf_token = os.environ.get("HF_TOKEN")
 
-if not hf_token:
-    raise ValueError("HF_TOKEN not found in environment variables. Please set it in Space Settings.")
-
-# --- LOGIC ---
 def load_pdf(file_path):
     loader = PyMuPDFLoader(file_path)
     docs = loader.load()
@@ -34,7 +32,6 @@ def ask_question(query, vector_store):
     retriever = vector_store.as_retriever()
     docs = retriever.get_relevant_documents(query)
     
-    # Using HuggingFaceEndpoint (newer) or Hub to call Mixtral
     llm = HuggingFaceHub(
         repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1",
         model_kwargs={"temperature": 0.7, "max_length": 512},
@@ -48,7 +45,6 @@ def ask_question(query, vector_store):
 def process_input(weblink, pdf_file, question):
     docs = []
     
-    # Error handling for empty inputs
     if not weblink and not pdf_file:
         return "Please provide a website link or upload a PDF."
     if not question:
@@ -58,7 +54,7 @@ def process_input(weblink, pdf_file, question):
         if weblink:
             docs.extend(load_website(weblink))
         if pdf_file:
-            docs.extend(load_pdf(pdf_file.name)) # Gradio handles file paths
+            docs.extend(load_pdf(pdf_file.name))
 
         vector_store = setup_vector_store(docs)
         response = ask_question(question, vector_store)
@@ -75,8 +71,7 @@ demo = gr.Interface(
         gr.Textbox(label="Ask a Question")
     ],
     outputs=gr.Textbox(label="Final Answer"),
-    title="Web & PDF QA System",
-    description="Upload a PDF or enter a website URL to chat with the content."
+    title="Web & PDF QA System"
 )
 
 if __name__ == "__main__":