updated

app.py

@@ -1,17 +1,21 @@
 import streamlit as st
 from langchain.prompts import PromptTemplate
-from langchain.llms import CTransformers
+from langchain_community.llms import CTransformers
 from src.helper import download_hf_embeddings, text_split, download_hf_model
-from langchain.vectorstores import Pinecone as LangchainPinecone
+from langchain_community.vectorstores import Pinecone as LangchainPinecone
 import os
 from dotenv import load_dotenv
 from src.prompt import prompt_template
 from langchain.chains import RetrievalQA
 import time
+from pinecone import Pinecone
+
 
 # Load environment variables
+load_dotenv()
 
 PINECONE_API_KEY = os.environ.get('PINECONE_API_KEY')
+index_name = "medicure-chatbot"
 
 # Set page configuration
 st.set_page_config(page_title="Medical Chatbot", page_icon="🏥", layout="wide")
@@ -62,14 +66,25 @@ st.title("🏥 Medical Chatbot")
 # Initialize the chatbot components
 @st.cache_resource
 def initialize_chatbot():
+
     embeddings = download_hf_embeddings()
-    model_name_or_path = "TheBloke/Llama-2-7B-Chat-GGML"
-    model_basename = "llama-2-7b-chat.ggmlv3.q4_0.bin"
-    model_path = download_hf_model(model_name_or_path, model_basename)
+    # model_name_or_path = "TheBloke/Llama-2-7B-Chat-GGML"
+    # model_basename = "llama-2-7b-chat.ggmlv3.q4_0.bin"
+    # model_path = download_hf_model(model_name_or_path, model_basename)
+    model_path = "TheBloke/Llama-2-7B-Chat-GGML"
     llm = CTransformers(model=model_path,
                         model_type="llama",
                         config={'max_new_tokens': 512,
                                 'temperature': 0.8})
+
+       
+    
+    
+    # initiaize pinecone
+    
+    pc = Pinecone(api_key=PINECONE_API_KEY)
+    index = pc.Index(index_name)
+
     PROMPT = PromptTemplate(template=prompt_template, input_variables=["context", "question"])
     chain_type_kwargs = {"prompt": PROMPT}
     docsearch = LangchainPinecone(index, embeddings.embed_query, "text")
@@ -114,11 +129,11 @@ def load_animation():
 st.markdown("""
 <div class="footer">
     <div class="social-icons">
-        <a href="https://github.com/yourusername" target="_blank"><i class="fab fa-github"></i></a>
-        <a href="https://linkedin.com/in/yourusername" target="_blank"><i class="fab fa-linkedin"></i></a>
-        <a href="https://yourwebsite.com" target="_blank"><i class="fas fa-globe"></i></a>
+        <a href="https://github.com/4darsh-Dev" target="_blank"><i class="fab fa-github"></i></a>
+        <a href="https://linkedin.com/in/adarsh-maurya-dev" target="_blank"><i class="fab fa-linkedin"></i></a>
+        <a href="https://adarshmaurya.onionreads.com" target="_blank"><i class="fas fa-globe"></i></a>
     </div>
-    <p>© 2024 Medical Chatbot. All rights reserved.</p>
+    <p>Adarsh Maurya© 2024 Medical Chatbot. All rights reserved.</p>
 </div>
 """, unsafe_allow_html=True)
 

src/helper.py

@@ -1,6 +1,6 @@
-from langchain.document_loaders import PyPDFDirectoryLoader
+from langchain_community.document_loaders import PyPDFDirectoryLoader
 from langchain.text_splitter import RecursiveCharacterTextSplitter
-from langchain.embeddings import HuggingFaceEmbeddings
+from langchain_community.embeddings import HuggingFaceEmbeddings
 from huggingface_hub import hf_hub_download