import os
import streamlit as st
from groq import Groq
from dotenv import load_dotenv

# Load environment variables from .env file
load_dotenv()

# Set up Groq client with API key
client = Groq(
    api_key=os.getenv("GROQ_API_KEY"),
)

# Streamlit UI
st.title("AI-powered CV Generator")

# Input fields for CV details
name = st.text_input("Name")
email = st.text_input("Email")
phone = st.text_input("Phone Number")
experience = st.text_area("Experience")
skills = st.text_area("Skills")

# Button to generate CV
if st.button("Generate CV"):
    # Constructing the content for the Groq model
    prompt = f"Create a CV for {name}. Email: {email}, Phone: {phone}. Experience: {experience}. Skills: {skills}."

    # Calling the Groq API
    chat_completion = client.chat.completions.create(
        messages=[
            {
                "role": "user",
                "content": prompt,
            }
        ],
        model="mixtral-8x7b-32768",
    )

    # Displaying the generated CV
    cv_content = chat_completion.choices[0].message.content
    st.write(cv_content)