Update server.py

server.py

@@ -1,9 +1,11 @@
 import os
 from utilities.transcripts import VTTTranscriptLoader, DocumentEmbedder
+from utilities.setup import get_files
+
 
 class EmbeddingService:
     def __init__(self, conf):
-        self.keys = self.get_keys()
+        self.keys = get_files.get_keys()
         self.conf = conf
 
     def __enter__(self):
@@ -12,14 +14,6 @@ class EmbeddingService:
 
     def __exit__(self, exc_type, exc_val, exc_tb):
         print("Exiting Embedding Service")
-
-    def get_keys(self):
-        "Get keys required to set up loading"
-        return {
-            "pinecone": os.environ["PINECONE_TOKEN"],
-            "huggingface": os.environ["HUGGINGFACE_TOKEN"], 
-            "openai": "", # leaving this alone for future use.
-        }
     
     def get_transcripts(self, files):
         # Get filepaths and load them in document format
@@ -48,8 +42,25 @@ class EmbeddingService:
 
 class QAService:
     def __init__(self, conf):
-        self.keys = get_keys()
+        self.keys = get_files.get_keys()
         self.conf = conf
 
+    def __enter__(self):
+        print("Start QA Service")
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        print("Exiting QA Service")
+
+    # First embed question
+
+    # Next, pass it into pinecone
+
+    # Third, retrieve pinecone search
+
+    # Structure query
+
+    # inference QA model
+    
     def run():
         return 0