Update server.py

server.py

@@ -1,9 +1,51 @@
+import os
+from utilities.transcripts import VTTTranscriptLoader, DocumentEmbedder
 
+class EmbeddingService:
+    def __init__(self, conf):
+        self.keys = self.get_keys()
+        self.conf = conf
 
-def submit_weights(model, repository, model_out_name, token):
-    """submits model to repository"""
-    repo = repository + '/' + model_out_name
+
+    def get_keys(self):
+        "Get keys required to set up loading"
+        return {
+            "pinecone": os.environ["PINECONE_TOKEN"],
+            "hugginface": os.environ["HUGGINFACE_TOKEN"], 
+            "openai": "", # leaving this alone for future use.
+        }
+    
+    
+    def get_transcripts(self, files):
+        # Get filepaths and load them in document format
+        filepaths = [file.name for file in files]
+        loader = VTTTranscriptLoader(file_path)
+        results = loader.load()
+            
+        return results
+    
     
-    model.push_to_hub(repo, token = token)
-    tokenizer.push_to_hub(repo, token = token)
-    return 0
+    def run(self, files):
+        # gets the files, cleans them, and loads them into pinecone
+        results = get_transcripts(files)
+        
+        doc_embedder =  DocumentEmbedder(
+            api_keys=self.keys,
+            files=results,
+            embedding=self.conf["embeddings"]["embedding"],
+            index_name=self.conf["embeddings"]["index_name"],
+        )
+        
+        # uploads them into pinecone
+        doc_embedder.embed()
+        
+        return 0
+
+
+class QAService:
+    def __init__(self, conf):
+        self.keys = get_keys()
+        self.conf = conf
+
+    def run():
+        return 0