Update app.py

app.py

@@ -7,16 +7,19 @@ openai.api_key = os.environ["OPENAI_API_KEY"]
 
 def translate_code(code, from_language, to_language):
     try:
-        prompt = f"Translate the following {from_language} code to {to_language}:\n{code}\n---\nTranslated code:\n"
+        context = f"Translate the following {from_language} code to {to_language}:\n{code}\n---\nTranslated code:\n"
+        
+        data = {
+            "model": "gpt-3.5-turbo",
+            "messages": [{"role": "system", "content": "You are a helpful assistant that translates code between different programming languages."}, {"role": "user", "content": context}],
+            "max_tokens": 200,
+            "temperature": 0.6,
+            "top_p": 1,
+            "frequency_penalty": 0,
+            "presence_penalty": 0,
+        }
 
-        response = openai.Completion.create(
-            engine="davinci-codex",
-            prompt=prompt,
-            max_tokens=200,
-            n=1,
-            stop=None,
-            temperature=0.5,
-        )
+        response = openai.Completion.create(**data)
 
         translated_code = response.choices[0].text.strip()
         return translated_code
@@ -44,7 +47,7 @@ iface = gr.Interface(
     ],
     outputs=gr.outputs.Textbox(label="Translated Code"),
     title="Code Translator",
-    description="Translate code between different programming languages using OpenAI.",
+    description="Translate code between different programming languages using OpenAI GPT-3.5 Turbo.",
 )
 
 iface.launch()