Upload 7 files

app.py

@@ -0,0 +1,50 @@
+from langchain.document_loaders import DirectoryLoader
+from langchain.text_splitter import RecursiveCharacterTextSplitter
+from langchain.embeddings import SentenceTransformerEmbeddings
+from langchain.vectorstores import Chroma
+from langchain_community.llms.huggingface_pipeline import HuggingFacePipeline
+from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
+from langchain.chains.question_answering import load_qa_chain
+import gradio as gr
+import torch 
+from huggingface_hub import login
+import os 
+
+directory = 'pets'
+HF_TOKEN = os.getenv("HF_TOKEN")
+login(token = HF_TOKEN)
+
+def load_docs(directory):
+  loader = DirectoryLoader(directory)
+  documents = loader.load()
+  return documents
+
+def split_docs(documents,chunk_size=1000,chunk_overlap=20):
+  text_splitter = RecursiveCharacterTextSplitter(chunk_size=chunk_size, chunk_overlap=chunk_overlap)
+  docs = text_splitter.split_documents(documents)
+  return docs
+
+documents = load_docs(directory)
+docs = split_docs(documents)
+embeddings = SentenceTransformerEmbeddings(model_name="thenlper/gte-large")
+
+db = Chroma.from_documents(docs, embeddings)
+
+model_id = "google/gemma-1.1-2b-it"
+tokenizer = AutoTokenizer.from_pretrained(model_id)
+model = AutoModelForCausalLM.from_pretrained(model_id,torch_dtype=torch.bfloat16)
+pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, max_new_tokens=100)
+hf = HuggingFacePipeline(pipeline=pipe)
+
+chain = load_qa_chain(hf, chain_type="stuff",verbose=True)
+
+def output(query, history):
+  
+    matching_docs = db.similarity_search(query)
+    answer =  chain.run(input_documents=matching_docs, question=query)
+    idx = answer.find("Answer")
+
+    return answer[idx:]
+
+
+gr.ChatInterface(output).launch()

pets/Different Types of Pet Animals.txt

@@ -0,0 +1 @@
+Pet animals come in all shapes and sizes, each suited to different lifestyles and home environments. Dogs and cats are the most common, known for their companionship and unique personalities. Small mammals like hamsters, guinea pigs, and rabbits are often chosen for their low maintenance needs. Birds offer beauty and song, and reptiles like turtles and lizards can make intriguing pets. Even fish, with their calming presence, can be wonderful pets.

pets/Health Care for Pets.txt

@@ -0,0 +1 @@
+Routine health care is crucial for pets to live long, happy lives. Regular vet check-ups help catch potential issues early and keep vaccinations up to date. Dental care is also essential to prevent diseases in pets, especially in dogs and cats. Regular grooming, parasite control, and weight management are other important aspects of pet health care.

pets/Nutrition Needs of Pet Animals.txt

@@ -0,0 +1 @@
+Proper nutrition is vital for the health and wellbeing of pets. Dogs and cats require a balanced diet that includes proteins, carbohydrates, and fats. Some may even have specific dietary needs based on their breed or age. Birds typically thrive on a diet of seeds, fruits, and vegetables, while reptiles have diverse diets ranging from live insects to fresh produce. Fish diets depend greatly on the species, with some needing live food and others subsisting on flakes or pellets.

pets/The Emotional Bond Between Humans and Pets.txt

@@ -0,0 +1 @@
+Pets offer more than just companionship; they provide emotional support, reduce stress, and can even help their owners lead healthier lives. The bond between pets and their owners is strong, and many people consider their pets as part of the family. This bond can be especially important in times of personal or societal stress, providing comfort and consistency.

pets/Training and Behaviour of Pets.txt

@@ -0,0 +1 @@
+Training is essential for a harmonious life with pets, particularly for dogs. It helps pets understand their boundaries and makes cohabitation easier for both pets and owners. Training should be based on positive reinforcement. Understanding pet behavior is also important, as changes in behavior can often be a sign of underlying health issues.

requirements.txt

@@ -0,0 +1,10 @@
+langchain
+transformers
+chroma
+datasets 
+pandas 
+chromadb 
+sentence_transformers
+transformers
+accelerate
+unstructured