added helper functions

src/helper.py

@@ -0,0 +1,64 @@
+from langchain.document_loaders import PyPDFDirectoryLoader
+from langchain.text_splitter import RecursiveCharacterTextSplitter
+from langchain.embeddings import HuggingFaceEmbeddings
+from huggingface_hub import hf_hub_download
+
+
+# loading the data
+def load_data(path):
+    loader = PyPDFDirectoryLoader(path)
+    extracted_data = loader.load()
+    return extracted_data
+
+
+#Create text chunks
+def text_split(extracted_data):
+    text_splitter = RecursiveCharacterTextSplitter(chunk_size = 500, chunk_overlap = 20)
+    text_chunks = text_splitter.split_documents(extracted_data)
+
+    return text_chunks
+
+
+#download embedding model
+def download_hf_embeddings():
+    embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
+    return embeddings
+
+
+# downloading any pdf on web
+
+import os
+import requests
+
+def download_pdf(url):
+    if not os.path.exists('data'):
+        os.makedirs('data')
+
+    pdf_url = url
+
+    # Get the filename from the URL
+    filename = pdf_url.split("/")[-1]
+
+    # Full path where the PDF will be saved
+    save_path = os.path.join('data', filename)
+
+    # Download the PDF
+    response = requests.get(pdf_url)
+
+    # Check if the request was successful
+    if response.status_code == 200:
+        # Write the content to a file
+        with open(save_path, 'wb') as file:
+            file.write(response.content)
+        print(f"PDF downloaded and saved to {save_path}")
+    else:
+        print(f"Failed to download PDF. Status code: {response.status_code}")
+
+
+
+
+
+def download_hf_model(model_name_or_path, model_basename):
+    model_path = hf_hub_download(repo_id=model_name_or_path, filename=model_basename)
+    return model_path
+

src/prompt.py

@@ -0,0 +1,19 @@
+prompt_template = """
+You are a knowledgeable medical AI assistant. Answer the user's health-related question based on the given context and your medical knowledge. Follow these guidelines:
+
+1. Provide clear, concise, and medically accurate information.
+2. Use the context provided, but interpret it with your medical expertise if needed.
+3. Explain medical terms in simple language when used.
+4. If the question involves diagnosis or treatment, advise consulting a healthcare professional.
+5. Acknowledge if you don't have enough information to answer fully.
+6. Never guess or make up medical information.
+7. Offer general health tips related to the question if appropriate.
+
+Context: {context}
+
+Question: {question}
+
+Remember: Your role is to inform, not diagnose. Prioritize accuracy and patient safety.
+
+Helpful answer:
+"""