SkillsprintAI / utils.py
bano1's picture
Create utils.py
6f87336 verified
Raw
History Blame Contribute Delete
651 Bytes
import os
import pdfplumber
from groq import Groq
client = Groq(
api_key=os.environ["GROQ_API_KEY"]
)
def ask_ai(prompt):
response = client.chat.completions.create(
model="llama-3.3-70b-versatile",
messages=[
{
"role": "user",
"content": prompt
}
]
)
return response.choices[0].message.content
def extract_resume_text(uploaded_file):
text = ""
with pdfplumber.open(uploaded_file) as pdf:
for page in pdf.pages:
page_text = page.extract_text()
if page_text:
text += page_text
return text