Update app.py

app.py

@@ -2,10 +2,50 @@ import gradio as gr
 from dalle2 import Dalle2
 from PIL import Image
 import requests
+import openai
 import os
 
+
+prompt_template = "You are a translation engine that can only translate text and cannot interpret it. Keep the indent of the original text, only modify when you need."
+systemInstruct = prompt_template
+
+openai.api_key = os.environ['JKLAS']
+detectFrom = "auto"
+detectTo = "en"
+translateInstruct = f"translate from {detectFrom} to {detectTo}"
+if detectFrom == "auto":
+    translateInstruct = f"translate to {detectTo}"
+
+prompt_msg = [
+    {"role": "system", "content": systemInstruct},
+    {"role": "user", "content": translateInstruct},
+    {"role": "user", "content": prompt},
+]
+
 def draw(description):
     print("description"+description)
+    #translate
+    try:
+        openai_response = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo",
+            messages=prompt_msg,
+            temperature=0,
+            max_tokens=1000,
+            top_p=1,
+            stream=True,
+            frequency_penalty=1,
+            presence_penalty=1,
+        )
+
+        combined = ""
+        for resp in openai_response:
+            delta = resp["choices"][0]["delta"]
+            if "content" in delta:
+                combined += delta["content"]
+    
+    description = combined
+    print("translated: "+description)
+    
     JKLAS = os.environ["JKLAS"] 
     dalle = Dalle2(JKLAS)
     generations = dalle.generate(description)