Update app.py

app.py

@@ -1,53 +1,45 @@
 import os
 import openai
 import gradio as gr
+import requests
+import json
 
-# Read the OpenAI API key from the environment variable
 openai.api_key = os.environ["OPENAI_API_KEY"]
 
 def translate_code(code, from_language, to_language):
+    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 programming languages."}, {"role": "user", "content": context}],
+        "max_tokens": 200,
+        "temperature": 0.6,
+        "top_p": 1,
+        "frequency_penalty": 0,
+        "presence_penalty": 0,
+    }
+
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {openai.api_key}",
+    }
+
     try:
-        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.ChatCompletion.create(**data)
-
-        translated_code = response.choices[0].text.strip()
-        return translated_code
-    except Exception as e:
-        return f"Error: {str(e)}"
-
-languages = [
-    "Python",
-    "Javascript",
-    "PHP",
-    "Java",
-    "Ruby",
-    "Golang",
-    "C#",
-    "SQL",
-    "Perl",
+        response = requests.post("https://api.openai.com/v1/chat/completions", headers=headers, data=json.dumps(data))
+        response.raise_for_status()
+        result = response.json()
+
+        return result["choices"][0]["message"]["content"].strip()
+
+    except requests.exceptions.HTTPError as e:
+        return f"Error: {str(e)}\nResponse: {response.text}"
+
+inputs = [
+    gr.inputs.Textbox(lines=100, 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"),
 ]
 
-iface = gr.Interface(
-    fn=translate_code,
-    inputs=[
-        gr.inputs.Textbox(lines=20, label="Original Code"),
-        gr.inputs.Dropdown(choices=languages, label="From Language"),
-        gr.inputs.Dropdown(choices=languages, label="To Language"),
-    ],
-    outputs=gr.outputs.Textbox(label="Translated Code"),
-    title="Code Translator",
-    description="Translate code between different programming languages using OpenAI GPT-3.5 Turbo.",
-)
-
-iface.launch()
+outputs = gr.outputs.Textbox(label="Translated Code")
+
+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()