Update app.py

app.py

@@ -7,41 +7,40 @@ import json
 openai.api_key = os.environ["OPENAI_API_KEY"]
 
 def translate_code(code, from_language, to_language):
-    # Define a message for each language pair to ensure the output can compile
-    if from_language == "Python" and to_language == "JavaScript":
-        context = f"Translate the following {from_language} code snippet to {to_language}:\n\n{code}\n\nEnsure that the translation follows JavaScript syntax and idioms, considering language-specific best practices. You may also need to add required imports/modules."
-    elif from_language == "JavaScript" and to_language == "Python":
-        context = f"Translate the following {from_language} code snippet to {to_language}:\n\n{code}\n\nEnsure that the translation follows Python syntax and idioms, considering language-specific best practices. You may also need to add required imports/modules."
-    elif from_language == "Java" and to_language == "C++":
-        context = f"Translate the following {from_language} code snippet to {to_language}:\n\n{code}\n\nEnsure that the translation follows C++ syntax and idioms, considering language-specific best practices. You may also need to add required includes/libraries."
+    if from_language == "JavaScript" and to_language == "Java":
+        context = f"Translate the following {from_language} code snippet to {to_language}:\n\n{code}\n\nEnsure that the translation follows Java syntax and idioms, considering language-specific best practices. You may also need to add required import statements."
+    elif from_language == "Java" and to_language == "JavaScript":
+        context = f"Translate the following {from_language} code snippet to {to_language}:\n\n{code}\n\nEnsure that the translation follows JavaScript syntax and idioms, considering language-specific best practices. You may also need to add required modules."
     elif from_language == "C++" and to_language == "Java":
-        context = f"Translate the following {from_language} code snippet to {to_language}:\n\n{code}\n\nEnsure that the translation follows Java syntax and idioms, considering language-specific best practices. You may also need to add required imports/packages."
+        context = f"Translate the following {from_language} code snippet to {to_language}:\n\n{code}\n\nEnsure that the translation follows Java syntax and idioms, considering language-specific best practices. You may also need to add required import statements."
+    elif from_language == "Java" and to_language == "C++":
+        context = f"Translate the following {from_language} code snippet to {to_language}:\n\n{code}\n\nEnsure that the translation follows C++ syntax and idioms, considering language-specific best practices. You may also need to include required headers."
     elif from_language == "C#" and to_language == "Java":
-        context = f"Translate the following {from_language} code snippet to {to_language}:\n\n{code}\n\nEnsure that the translation follows Java syntax and idioms, considering language-specific best practices. You may also need to add required imports/packages."
+        context = f"Translate the following {from_language} code snippet to {to_language}:\n\n{code}\n\nEnsure that the translation follows Java syntax and idioms, considering language-specific best practices. You may also need to add required import statements."
     elif from_language == "Java" and to_language == "C#":
         context = f"Translate the following {from_language} code snippet to {to_language}:\n\n{code}\n\nEnsure that the translation follows C# syntax and idioms, considering language-specific best practices. You may also need to add required using statements."
-    elif from_language == "Ruby" and to_language == "Python":
-        context = f"Translate the following {from_language} code snippet to {to_language}:\n\n{code}\n\nEnsure that the translation follows Python syntax and idioms, considering language-specific best practices. You may also need to add required imports/modules."
-    elif from_language == "Python" and to_language == "Ruby":
+    elif from_language == "Ruby" and to_language == "Java":
+        context = f"Translate the following {from_language} code snippet to {to_language}:\n\n{code}\n\nEnsure that the translation follows Java syntax and idioms, considering language-specific best practices. You may also need to add required import statements."
+    elif from_language == "Java" and to_language == "Ruby":
         context = f"Translate the following {from_language} code snippet to {to_language}:\n\n{code}\n\nEnsure that the translation follows Ruby syntax and idioms, considering language-specific best practices. You may also need to add required gems/libraries."
     elif from_language == "Swift" and to_language == "Python":
-        context = f"Translate the following {from_language} code snippet to {to_language}:\n\n{code}\n\nEnsure that the translation follows Python syntax and idioms, considering language-specific best practices. You may also need to add required imports/modules."
+        context = f"Translate the following {from_language} code snippet to {to_language}:\n\n{code}\n\nEnsure that the translation follows Python syntax and idioms, considering language-specific best practices. You may also need to add required modules."
     elif from_language == "Python" and to_language == "Swift":
-        context = f"Translate the following {from_language} code snippet to {to_language}:\n\n{code}\n\nEnsure that the translation follows Swift syntax and idioms, considering language-specific best practices. You may also need to add required imports/frameworks."
-    elif from_language == "Go" and to_language == "Python":
-        context = f"Translate the following {from_language} code snippet to {to_language}:\n\n{code}\n\nEnsure that the translation follows Python syntax and idioms, considering language-specific best practices. You may also need to add required imports/modules."
-    elif from_language == "Python" and to_language == "Go":
-        context = f"Translate the following {from_language} code snippet to {to_language}:\n\n{code}\n\nEnsure that the translation follows Go syntax and idioms, considering language-specific best practices. You may also need to add required imports/packages."
-    elif from_language == "PHP" and to_language == "Python":
-        context = f"Translate the following {from_language} code snippet to {to_language}:\n\n{code}\n\nEnsure that the translation follows Python syntax and idioms, considering language-specific best practices. You may also need to add required imports/modules."
-    elif from_language == "Python" and to_language == "PHP":
-        context = f"Translate the following {from_language} code snippet to {to_language}:\n\n{code}\n\nEnsure that the translation follows PHP syntax and idioms, considering language-specific best practices. You may also need to add required includes/libraries."
+        context = f"Translate the following {from_language} code snippet to {to_language}:\n\n{code}\n\nEnsure that the translation follows Swift syntax and idioms, considering language-specific best practices. You may also need to add required frameworks."
+    elif from_language == "Go" and to_language == "Java":
+        context = f"Translate the following {from_language} code snippet to {to_language}:\n\n{code}\n\nEnsure that the translation follows Java syntax and idioms, considering language-specific best practices. You may also need to add required import statements."
+    elif from_language == "Java" and to_language == "Go":
+        context = f"Translate the following {from_language} code snippet to {to_language}:\n\n{code}\n\nEnsure that the translation follows Go syntax and idioms, considering language-specific best practices. You may also need to add required packages."
+    elif from_language == "PHP" and to_language == "Java":
+        context = f"Translate the following {from_language} code snippet to {to_language}:\n\n{code}\n\nEnsure that the translation follows Java syntax and idioms, considering language-specific best practices. You may also need to add required import statements."
+    elif from_language == "Java" and to_language == "PHP":
+        context = f"Translate the following {from_language} code snippet to {to_language}:\n\n{code}\n\nEnsure that the translation follows PHP syntax and idioms, considering language-specific best practices. You may also need to add required packages/libraries."
     else:
-        context = f"Translate the following {from_language} code snippet to {to_language}:\n\n{code}\n\nEnsure that the translation is accurate and idiomatic, considering language-specific best practices."
-
+        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}],
+        "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,
@@ -55,13 +54,12 @@ def translate_code(code, from_language, to_language):
     }
 
     try:
-        response = requests.post("https://api.openai.com/v1/chat/completions", headers=headers, data=json.dumps(data
-   ))
+        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()
-
+        return result["choices"][0]["message"]["content"].
+strip()
     except requests.exceptions.HTTPError as e:
         return f"Error: {str(e)}\nResponse: {response.text}"
 
@@ -73,4 +71,4 @@ inputs = [
 
 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()
+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, along with specific instructions for each language pair to ensure that the code can be compiled without issues.", theme="compact").launch()