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()