Adding OPEN AI calls

app.py

@@ -5,10 +5,12 @@ from typing import List, Tuple, Optional, Dict, Union
 import google.generativeai as genai
 import gradio as gr
 from PIL import Image
+from openai import OpenAI
 
 print("google-generativeai:", genai.__version__)
 
 GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
+client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
 
 TITLE = """<h1 align="center">Gemini PolyGLOT that can understand languages beyond english</h1>"""
 SUBTITLE = """"""
@@ -32,6 +34,21 @@ IMAGE_CACHE_DIRECTORY = "/tmp"
 IMAGE_WIDTH = 512
 CHAT_HISTORY = List[Tuple[Optional[Union[Tuple[str], str]], Optional[str]]]
 
+def translate_to_english(text):
+    response = client.chat.completions.create(model = "gpt-3.5-turbo",
+    messages = [
+        {"role": "system", "content": "You are a translator to English. Check if the input is in English. If the input is english DO NOTHING and return AS-IS else Translate to English"},
+        {"role": "user", "content": text},
+    ],stream = False, )
+    return response.choices[0].message.content
+
+def translate_from_english(text, language):
+    response = client.chat.completions.create(model = "gpt-3.5-turbo",
+    messages = [
+        {"role": "system", "content": "You are a translator to " + language + " Translate english to " + language},
+        {"role": "user", "content": text},
+    ],stream = False, )
+    return response.choices[0].message.content
 
 def preprocess_stop_sequences(stop_sequences: str) -> Optional[List[str]]:
     if not stop_sequences:
@@ -89,7 +106,8 @@ def bot(
     stop_sequences: str,
     top_k: int,
     top_p: float,
-    chatbot: CHAT_HISTORY
+    chatbot: CHAT_HISTORY,
+    language: str
 ):
     if len(chatbot) == 0:
         return chatbot
@@ -122,8 +140,11 @@ def bot(
         text_prompt = [chatbot[-1][0]] \
             if chatbot[-1][0] and isinstance(chatbot[-1][0], str) \
             else []
-        
+        if (language.startswith("en") == False):
+            text_prompt = [translate_to_english(text_prompt[0])]
         model = genai.GenerativeModel('gemini-pro')
+
+
         response = model.generate_content(
             text_prompt,
             stream=True,
@@ -223,6 +244,7 @@ user_inputs = [
     text_prompt_component,
     chatbot_component
 ]
+language = gr.Textbox(label="description", value = "en")
 
 bot_inputs = [
     google_key_component,
@@ -232,13 +254,13 @@ bot_inputs = [
     stop_sequences_component,
     top_k_component,
     top_p_component,
-    chatbot_component
+    chatbot_component,
+    language
 ]
 
 with gr.Blocks() as demo:
     gr.HTML(TITLE)
     gr.HTML(SUBTITLE)
-    language = gr.Textbox(label="description", value = "en")
     with gr.Column():
         google_key_component.render()
         chatbot_component.render()

requirements.txt

@@ -2,3 +2,4 @@ google-generativeai==0.3.1
 gradio
 googletrans==3.1.0a0
 trulens_eval==0.19.2
+openai===1.6.1