Update app.py

app.py

@@ -4,11 +4,13 @@ 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
 from langchain_community.vectorstores import FAISS
-from langchain_huggingface import HuggingFaceEndpoint
 from langchain.chains.question_answering import load_qa_chain
 
+# --- THE NEW MODERN IMPORTS ---
+# These replace the old tools that were causing the errors
+from langchain_huggingface import HuggingFaceEndpoint, HuggingFaceEmbeddings
+
 # Get the token from the secrets
 hf_token = os.environ.get("HF_TOKEN")
 
@@ -23,6 +25,7 @@ def load_website(url):
     return docs
 
 def setup_vector_store(docs):
+    # use the new HuggingFaceEmbeddings class
     embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
     vector_store = FAISS.from_documents(docs, embeddings)
     return vector_store
@@ -31,12 +34,14 @@ def ask_question(query, vector_store):
     retriever = vector_store.as_retriever()
     docs = retriever.get_relevant_documents(query)
     
-    # Using HuggingFaceEndpoint with specific task parameters
+    # Use the new HuggingFaceEndpoint
+    # This automatically handles the connection without the "post" error
     llm = HuggingFaceEndpoint(
         repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1",
         task="text-generation",
         max_new_tokens=512,
-        temperature=0.5,
+        do_sample=True,
+        temperature=0.7,
         huggingfacehub_api_token=hf_token
     )
     
@@ -58,6 +63,9 @@ def process_input(weblink, pdf_file, question):
         if pdf_file:
             docs.extend(load_pdf(pdf_file.name))
 
+        if not docs:
+            return "Could not load any content."
+
         vector_store = setup_vector_store(docs)
         response = ask_question(question, vector_store)
         return response
@@ -73,7 +81,8 @@ demo = gr.Interface(
         gr.Textbox(label="Ask a Question")
     ],
     outputs=gr.Textbox(label="Final Answer"),
-    title="Web & PDF QA System"
+    title="Web & PDF QA System",
+    description="Ask questions about any website or PDF document."
 )
 
 if __name__ == "__main__":