openai key

app.py

@@ -5,15 +5,15 @@ from gradio_pdf import PDF
 
 import resume_parser
 
-def parse_resume(pdf):
-    resume = resume_parser.run(pdf)
+def parse_resume(pdf, use_openai=False, openai_key=""):
+    resume = resume_parser.run(pdf, use_openai, openai_key)
     if isinstance(resume, resume_parser.Resume):
         resume = resume.model_dump_json()
     return resume
 
 demo = gr.Interface(
     fn=parse_resume,
-    inputs=[PDF(label="Upload a PDF", interactive=True)],
+    inputs=[PDF(label="Upload a PDF", interactive=True), gr.Checkbox(label="Use OpenAI"), gr.Text(label="OpenAI Token")],
     outputs=[gr.JSON()],
     examples=[[f"samples/{fname}"] for fname in os.listdir("samples") if fname.endswith(".pdf")]
 )

requirements.txt

@@ -6,4 +6,5 @@ tiktoken
 huggingface_hub
 langchain
 langchain-community
+langchain-openai
 python-dotenv

resume_parser.py

@@ -1,3 +1,4 @@
+import os
 from json import JSONDecodeError
 
 import PyPDF2
@@ -6,7 +7,7 @@ from dotenv import load_dotenv
 from langchain.output_parsers import PydanticOutputParser
 from langchain.prompts import PromptTemplate
 from langchain_community.callbacks import get_openai_callback
-from langchain_community.chat_models import ChatOpenAI
+from langchain_openai import ChatOpenAI
 from langchain_community.llms import HuggingFaceEndpoint
 from langchain_core.messages import BaseMessage
 from pydantic import ValidationError
@@ -148,10 +149,11 @@ def extract_resume_fields(full_text, model):
         return output
 
 
-def run(pdf_file_path):
-    use_openai = False
+def run(pdf_file_path, use_openai=False, openai_key=""):
+    if len(openai_key) == 0 and "OPENAI_API_KEY" in os.environ:
+        openai_key = os.environ["OPENAI_API_KEY"]
     if use_openai:
-        model = ChatOpenAI(temperature=0, model="gpt-3.5-turbo", max_tokens=2500)
+        model = ChatOpenAI(temperature=0, model="gpt-3.5-turbo", max_tokens=2500, openai_api_key=openai_key)
     else:
         model = HuggingFaceEndpoint(repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1", temperature= 0.001, max_new_tokens=4000)