rca123456 commited on
Commit
a7cb257
·
verified ·
1 Parent(s): 6af8a1b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -8
app.py CHANGED
@@ -7,18 +7,22 @@ from langchain_core.output_parsers import StrOutputParser
7
  from langchain_groq import ChatGroq
8
  from sklearn.metrics.pairwise import cosine_similarity
9
  import numpy as np
 
 
 
10
  import os
 
 
11
 
12
- # ✅ Set Groq API Key
13
- os.environ["GROQ_API_KEY"] = "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" # Replace with your actual Groq API key
14
 
15
  def extract_text_from_pdf(pdf_file):
16
- temp_path = f"temp_{pdf_file.name}"
17
- with open(temp_path, "wb") as f:
18
- f.write(pdf_file.read())
19
- loader = PyPDFLoader(temp_path)
20
- pages = loader.load_and_split()
21
- os.remove(temp_path)
22
  return " ".join([page.page_content for page in pages])
23
 
24
  def extract_skills(text):
 
7
  from langchain_groq import ChatGroq
8
  from sklearn.metrics.pairwise import cosine_similarity
9
  import numpy as np
10
+ from dotenv import load_dotenv
11
+ load_dotenv()
12
+
13
  import os
14
+ groq_api_key = os.getenv("GROQ_API_KEY")
15
+ os.environ["GROQ_API_KEY"] = groq_api_key
16
 
17
+ import os
18
+ import tempfile
19
 
20
  def extract_text_from_pdf(pdf_file):
21
+ with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as temp:
22
+ temp.write(pdf_file)
23
+ temp.flush()
24
+ loader = PyPDFLoader(temp.name)
25
+ pages = loader.load_and_split()
 
26
  return " ".join([page.page_content for page in pages])
27
 
28
  def extract_skills(text):