Update app.py

app.py

@@ -21,10 +21,11 @@ from tqdm.auto import tqdm
 FILE_PATH = "anjibot_chunks.json"
 BATCH_SIZE = 384
 INDEX_NAME = "groq-llama-3-rag"
-PINECONE_API_KEY = os.getenv("PINECONE_API_KEY")  # Fixed syntax here
-GROQ_API_KEY = os.getenv("GROQ_API_KEY")  # Fixed s
+PINECONE_API_KEY = os.getenv("PINECONE_API_KEY")
+GROQ_API_KEY = os.getenv("GROQ_API_KEY") 
 DIMS = 768
 encoder = SentenceTransformer('dwzhu/e5-base-4k')
+groq_client = Groq(api_key=GROQ_API_KEY)
 
 with open(FILE_PATH, 'r') as file:
         data= json.load(file)
@@ -65,7 +66,7 @@ def get_docs(query: str, top_k: int) -> list[str]:
     res = index.query(vector=xq.tolist(), top_k=top_k, include_metadata=True)
     return [x["metadata"]['content'] for x in res["matches"]]
 
-def get_response(query: str, docs: list[str], groq_client: any) -> str:
+def get_response(query: str, docs: list[str]) -> str:
     system_message = (
         "You are Anjibot, the AI course rep of 400 Level Computer Science department. You are always helpful, jovial, can be sarcastic but still sweet.\n"
         "Provide the answer to class-related queries using\n"
@@ -90,14 +91,11 @@ def get_response(query: str, docs: list[str], groq_client: any) -> str:
 
 def handle_query(user_query: str):
 
-    # Initialize Groq client
-    groq_client = Groq(api_key=GROQ_API_KEY)
-
     # Get relevant documents
     docs = get_docs(user_query, top_k=5)
 
     # Generate and return response
-    response = get_response(user_query, docs, groq_client)
+    response = get_response(user_query, docs=docs)
 
     for word in response.split():
             yield word + " "