Update app.py

app.py

@@ -2,6 +2,7 @@ import subprocess
 subprocess.run('pip install -r requirements.txt', shell = True)
 
 import gradio as gr
+import cv2
 import os
 from PIL import Image
 import numpy as np
@@ -14,7 +15,8 @@ from langchain_community.embeddings import HuggingFaceEmbeddings
 from langchain.text_splitter import CharacterTextSplitter
 from langchain_core.output_parsers import StrOutputParser
 from langchain_core.runnables import RunnablePassthrough
-from langchain_fireworks import ChatFireworks
+#from langchain_fireworks import ChatFireworks
+from langchain_anthropic import ChatAnthropic
 from langchain_community.llms import Ollama
 from langchain_core.prompts import ChatPromptTemplate
 from rich.console import Console
@@ -74,19 +76,19 @@ def ragChain():
 
     output: rag chain
     """
-    loader = TextLoader("knowledgeBase.txt")
-    docs = loader.load()
+    #loader = TextLoader("knowledgeBase.txt")
+    #docs = loader.load()
 
-    text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
-    docs = text_splitter.split_documents(docs)
+    #text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
+    #docs = text_splitter.split_documents(docs)
 
-    vectorstore = FAISS.from_documents(documents = docs, embedding = HuggingFaceEmbeddings())
+    vectorstore = FAISS.load_local("faiss_index", embeddings = HuggingFaceEmbeddings(), allow_dangerous_deserialization = True)
     retriever = vectorstore.as_retriever(search_type = "similarity", search_kwargs = {"k": 5})
 
-    APIKEY = "o7T3gVx9Vt8GSJbLyPV1974vF8LXVp01CWqOkWQuHgoHm07H"
-    os.environ["FIREWORKS_API_KEY"] = APIKEY
+    #APIKEY = "o7T3gVx9Vt8GSJbLyPV1974vF8LXVp01CWqOkWQuHgoHm07H"
+    #os.environ["FIREWORKS_API_KEY"] = APIKEY
 
-    llm = ChatFireworks(model="accounts/fireworks/models/mixtral-8x7b-instruct")
+    llm = ChatAnthropic(model='claude-3-opus-20240229')
 
     prompt = ChatPromptTemplate.from_messages(
         [
@@ -98,17 +100,17 @@ def ragChain():
             ),
             (
                 "human",
-                """Provide information about the leaf disease in question in bullet points. 
+                """Provide information about the leaf disease in question in bullet points.
                 Start your answer by mentioning the disease (if any) or healthy in this format: 'Condition: disease name'.
                 """,
             ),
-            
+
             ("human", "{context}, {question}"),
         ]
     )
 
     rag_chain = (
-    { 
+    {
     "context": retriever,
     "question": RunnablePassthrough()
     }