Update app.py

app.py

@@ -7,11 +7,41 @@ import json
 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."
-    
+    # 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."
+    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."
+    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."
+    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":
+        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."
+    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."
+    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."
+
     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,
@@ -25,7 +55,8 @@ 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()
 
@@ -42,4 +73,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.", theme="compact").launch()