Update app.py

app.py

@@ -1,48 +1,38 @@
 import os
 import gradio as gr
-import google.generativeai as palm
+import openai
 
 # Retrieve the API key from the environment variable
-api_key = os.getenv('GENERATIVE_AI_API_KEY')
+api_key = os.getenv('OPENAI_API_KEY')
 if api_key is None:
-    raise Exception("Missing API key for Generative AI")
+    raise Exception("Missing API key for OpenAI")
 
-palm.configure(api_key=api_key)
-
-defaults = {
-  'model': 'models/chat-bison-001',
-  'temperature': 0.25,
-  'candidate_count': 1,
-  'top_k': 40,
-  'top_p': 0.95,
-}
+openai.api_key = api_key
 
 def generate_description(title, location, desired_experience, preferred_experience, about_the_team):
-    context = ""
-    examples = []
     company_name = "Imaginary Inc."  # Placeholder company name
     company_description = "Imaginary Inc. is a forward-thinking company that values innovation, creativity, and diversity. We believe in fostering a positive work environment where every employee can thrive."  # Company branding message
 
     messages = [
-        f"Company: {company_name}",
-        f"About the Company: {company_description}",
-        f"Job Title: {title}",
-        f"Job Location: {location}",
-        f"Desired Candidate Experience: {desired_experience}",
-        f"Preferred Candidate Experience: {preferred_experience}",
-        f"About the Team: {about_the_team}",
-        "NEXT REQUEST",
-        f"Generate a job description for the position {title} at {company_name}. The job is located in {location}. The desired experience for this role is {desired_experience} and the preferred experience is {preferred_experience}. The role is part of the following team: {about_the_team}. The description should align with the company's branding message: {company_description}"
+        {"role": "system", "content": "You are a helpful assistant."},
+        {"role": "user", "content": f"Company: {company_name}"},
+        {"role": "user", "content": f"About the Company: {company_description}"},
+        {"role": "user", "content": f"Job Title: {title}"},
+        {"role": "user", "content": f"Job Location: {location}"},
+        {"role": "user", "content": f"Desired Candidate Experience: {desired_experience}"},
+        {"role": "user", "content": f"Preferred Candidate Experience: {preferred_experience}"},
+        {"role": "user", "content": f"About the Team: {about_the_team}"},
+        {"role": "user", "content": f"Generate a job description for the position {title} at {company_name}. The job is located in {location}. The desired experience for this role is {desired_experience} and the preferred experience is {preferred_experience}. The role is part of the following team: {about_the_team}. The description should align with the company's branding message: {company_description}"}
     ]
-    
-    response = palm.chat(
-        **defaults,
-        context=context,
-        examples=examples,
-        messages=messages
+
+    response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=messages,
+        temperature=0.5,
+        max_tokens=500
     )
     
-    return response.last
+    return response['choices'][0]['message']['content']
 
 iface = gr.Interface(
     fn=generate_description,