Update app.py

app.py

@@ -1,18 +1,18 @@
-import openai
 import os
+import openai
 import gradio as gr
 import requests
 import json
 
-openai.api_key = "your_openai_api_key"
+openai.api_key = os.environ["OPENAI_API_KEY"]
 
 def translate_code(code, from_language, to_language):
     context = f"Translate the following {from_language} code snippet to {to_language}:\n\n{code}\n\nKeep the translation accurate and idiomatic, considering language-specific best practices."
-
+    
     data = {
         "model": "gpt-3.5-turbo",
-        "messages": [{"role": "system", "content": "You are a code translator assistant that translates code snippets between programming languages."}, {"role": "user", "content": context}],
-        "max_tokens": 300,
+        "messages":  [{"role": "system", "content": "You are a code translator assistant that translates code snippets between programming languages."}, {"role": "user", "content": context}],
+        "max_tokens": 200,
         "temperature": 0.6,
         "top_p": 1,
         "frequency_penalty": 0,
@@ -35,11 +35,11 @@ def translate_code(code, from_language, to_language):
         return f"Error: {str(e)}\nResponse: {response.text}"
 
 inputs = [
-    gr.inputs.Textarea(placeholder="Enter your code here...", label="Code"),
-    gr.inputs.Dropdown(choices=["Python", "JavaScript", "Java", "C++", "C#", "Ruby", "Go", "PHP"], label="From Language"),
-    gr.inputs.Dropdown(choices=["Python", "JavaScript", "Java", "C++", "C#", "Ruby", "Go", "PHP"], label="To Language"),
+    gr.inputs.Textbox(lines=10, placeholder="Enter your code here...", label="Code"),
+    gr.inputs.Dropdown(choices=["Python", "JavaScript", "Java", "C++", "C#", "Ruby", "Swift", "Go", "PHP"], label="From Language"),
+    gr.inputs.Dropdown(choices=["Python", "JavaScript", "Java", "C++", "C#", "Ruby", "Swift", "Go", "PHP"], label="To Language"),
 ]
 
-outputs = gr.outputs.Textarea(label="Translated Code")
+outputs = gr.outputs.Textbox(label="Translated Code")
 
-gr.Interface(fn=translate_code, inputs=inputs, outputs=outputs, title="Code Translator", description="Translate code between different programming languages. Note that the translations might not always be perfect and may require some manual adjustments.", theme="compact").launch()
+gr.Interface(fn=translate_code, inputs=inputs, outputs=outputs, title="Code Translator", description="Enter your code, select the source language, and the language you want to translate it to. The AI will try to provide the translated code.", theme="compact").launch()