Update app.py

app.py

@@ -3,9 +3,9 @@ import logging
 import gradio as gr
 import asyncio
 from dotenv import load_dotenv
-from langchain.document_loaders import ArxivLoader
-from langchain.vectorstores import Chroma
-from langchain_community.embeddings import HuggingFaceHubEmbeddings
+from langchain_community.document_loaders import ArxivLoader  # Updated import
+from langchain_community.vectorstores import Chroma  # Updated import
+from langchain_huggingface import HuggingFaceEmbeddings  # Updated import
 from langchain_groq import ChatGroq
 from PyPDF2 import PdfReader
 from huggingface_hub import login
@@ -24,14 +24,11 @@ if not HUGGING_API_KEY or not GROQ_API_KEY:
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# Authenticate with Hugging Face
+# Authenticate with Hugging Face (for model downloads)
 login(HUGGING_API_KEY)
 
-# Load models and embeddings with a specific embedding model
-embedding_model = HuggingFaceHubEmbeddings(
-    huggingfacehub_api_token=HUGGING_API_KEY,
-    model_id="sentence-transformers/all-MiniLM-L6-v2"  # Explicitly set a proper embedding model
-)
+# Load models and embeddings with a local embedding model
+embedding_model = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
 llm = ChatGroq(temperature=0, model_name="llama3-70b-8192", api_key=GROQ_API_KEY)
 client = AsyncGroq(api_key=GROQ_API_KEY)
 
@@ -39,7 +36,7 @@ client = AsyncGroq(api_key=GROQ_API_KEY)
 pdf_vector_store = None
 current_pdf_path = None
 
-# General Chat (unchanged)
+# General Chat
 async def chat_with_replit(message, history):
     try:
         messages = [{"role": "system", "content": "You are an assistant answering user questions."}]
@@ -59,7 +56,7 @@ async def chat_with_replit(message, history):
 def chat_with_replit_sync(message, history):
     return asyncio.run(chat_with_replit(message, history))
 
-# ArXiv Chat (unchanged)
+# ArXiv Chat
 async def chat_with_replit_arxiv(message, history, doi_num):
     try:
         loader = ArxivLoader(query=str(doi_num), load_max_docs=10)
@@ -125,7 +122,6 @@ def process_pdf(pdf_file):
 
 # Gradio UI
 with gr.Blocks() as app:
-    # General Chat (unchanged)
     with gr.Tab(label="General Chat"):
         gr.Markdown("### Chat with the Assistant")
         with gr.Row():
@@ -151,7 +147,6 @@ with gr.Blocks() as app:
         general_send_button.click(update_general_response, inputs=general_chat_history,
                                   outputs=[general_chat_history, general_chat_output])
 
-    # ArXiv Chat (unchanged)
     with gr.Tab(label="Chat with ArXiv Paper"):
         gr.Markdown("### Ask Questions About an ArXiv Paper")
         with gr.Row():
@@ -178,7 +173,6 @@ with gr.Blocks() as app:
         arxiv_send_button.click(update_arxiv_response, inputs=[arxiv_chat_history, arxiv_doi],
                                 outputs=[arxiv_chat_history, arxiv_chat_output])
 
-    # Local PDF Chat
     with gr.Tab(label="Chat with Local PDF"):
         gr.Markdown("### Ask Questions About an Uploaded PDF")
         pdf_file_input = gr.File(label="Upload PDF file", file_types=[".pdf"])
@@ -207,5 +201,4 @@ with gr.Blocks() as app:
         pdf_send_button.click(update_pdf_response, inputs=pdf_chat_history,
                               outputs=[pdf_chat_history, pdf_chat_output])
 
-app.launch()
-
+app.launch()