JD_Parser1 / app.py
KunaalNaik's picture
Update app.py
f8decbd verified
import streamlit as st
import google.generativeai as genai
# Header for the Streamlit app
st.header("Google Gemini - Generate Content")
# Retrieve the API key from Streamlit secrets
GOOGLE_API_KEY = st.secrets["GEMINI_API_KEY"]
# Configure the Google Generative AI API with your API key
genai.configure(api_key=GOOGLE_API_KEY)
# Text input for user prompt
user_input = st.text_area("Enter your prompt to generate content:")
prompt = f"""
Analyze the following Job Description:
1. Extract the top keywords.
2. Suggest a learning plan based on the skills required.
3. Provide the top interview questions a candidate might face.
Job Description: {user_input}
"""
# Button to submit the prompt
if st.button("Generate"):
if user_input:
# Initialize the model
model = genai.GenerativeModel('gemini-pro') # Assuming this is the correct model
try:
# Generate content based on the user's input
response = model.generate_content(prompt)
# Display the generated content
st.write("Generated Content:")
st.write(response.text)
except Exception as e:
st.error(f"Error: {e}")
else:
st.error("Please enter a prompt.")