Improve code

tinkerhub · Nov 23, 2023 · b5baf8c · b5baf8c
1 parent 41e6e4c
commit b5baf8c
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 5 deletions.
diff --git a/ops/.env b/ops/.env
@@ -0,0 +1 @@
+OPENAI_API_KEY=""
diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,5 @@
 openai
 sounddevice
 numpy
-langchain
+langchain
+python-dotenv
diff --git a/src/ai.py b/src/ai.py
@@ -1,12 +1,9 @@
 import os
-import openai
 from openai import OpenAI
 from utils import generate_text_with_prompt
 from langchain.document_loaders import DirectoryLoader
 
 
-openai.api_key = 'your-api-key'
-
 def generate_transcript(pdf_file, client: OpenAI):
     prompt = "Generate a full transcript for an event. \
         The transcript should start with a greeting and welcome to the event message. \

diff --git a/src/main.py b/src/main.py
@@ -1,9 +1,15 @@
 import argparse
 import ai
+import os
 import audio
+import dotenv
 from openai import OpenAI
 
-client = OpenAI()
+dotenv.load_dotenv("ops/.env")
+
+client = OpenAI(
+    api_key=os.environ.get("OPENAI_API_KEY"),
+)
 
 def main():
     parser = argparse.ArgumentParser(description='AI Event Host')