Update app.py

app.py

@@ -2,7 +2,7 @@ import streamlit as st
 import google.generativeai as genai
 
 # Header for the Streamlit app
-st.header("JD Keyword Extractor")
+st.header("Google Gemini - Generate Content")
 
 # Retrieve the API key from Streamlit secrets
 GOOGLE_API_KEY = st.secrets["GEMINI_API_KEY"]
@@ -10,24 +10,22 @@ 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 Job Description
-user_input = st.text_area("Enter the Job Description:")
+# Text input for user prompt
+user_input = st.text_area("Enter your prompt to generate content:")
 
-# Button to submit the Job Description
-if st.button("Extract Keywords"):
+# 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:
-            # Send the Job Description to the API, asking it to extract keywords
-            prompt = f"Extract the top keywords from this job description: {user_input}"
-            response = genai.generate_content(model="models/gemini-1.5-bison", prompt=prompt)
+            # Generate content based on the user's input
+            response = model.generate_content(user_input)
             
-            # Extract keywords from the response
-            keywords = response['candidates'][0]['output']
-
-            # Display the extracted keywords
-            st.write("Top Keywords:")
-            st.write(keywords)
+            # 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 Job Description.")
+        st.error("Please enter a prompt.")