Update app.py

app.py

@@ -1,26 +1,33 @@
-import openai
 import os
 import gradio as gr
+import google.generativeai as palm
 
-openai.api_key = os.environ["OPENAI_API_KEY"]
+palm.configure(api_key=os.environ["YOUR_API_KEY"])
+
+defaults = {
+ 'model': 'models/chat-bison-001',
+ 'temperature': 0.25,
+ 'candidate_count': 1,
+ 'top_k': 40,
+ 'top_p': 0.95,
+}
 
 def get_recommendation(prompt, recommendation_type):
+ context = ""
  if recommendation_type == "Letter of Recommendation":
  context = f"Write a formal Letter of Recommendation with a header and signature for a job addressed to hiring manager for a candidate who is male/he named Ziga who is a recruiting professional without mention of time worked together and follow this instructions: {prompt}"
  else:
  context = f"Write a Recommendation for LinkedIn for a candidate who is male/he named Ziga who is a recruiting professional without mention of time worked together and follow this instructions: {prompt}"
  
- response = openai.Completion.create(
- engine="text-davinci-003",
- prompt=context,
- temperature=0.6,
- max_tokens=3000,
- top_p=1,
- frequency_penalty=0,
- presence_penalty=0
+ messages = [context]
+ response = palm.chat(
+ **defaults,
+ context=context,
+ examples=[],
+ messages=messages
  )
 
- return response.choices[0].text.strip()
+ return response.last
 
 inputs = [
  gr.inputs.Radio(choices=["Letter of Recommendation", "LinkedIn Recommendation"], label="Recommendation Type"),
@@ -29,4 +36,4 @@ inputs = [
 
 outputs = gr.outputs.Textbox(label="Reply")
 
-gr.Interface(fn=get_recommendation, inputs=inputs, outputs=outputs, title="Recommendation Generator", description="Describe what you enjoyed and observed while working with Ziga. Choose the type of recommendation you want.", theme="compact").launch()
+gr.Interface(fn=get_recommendation, inputs=inputs, outputs=outputs, title="Recommendation Generator", description="Describe what you enjoyed and observed while working with Ziga. Choose the type of recommendation you want.", theme="compact").launch()