updated Chat completion for openai

app.py

@@ -159,12 +159,32 @@ with tab3:
             # Get 10 samples from the dataset for translation
             samples_to_translate = st.session_state.data.sample(10)['sentence'].tolist()
 
-            # Perform automatic translation using OpenAI GPT-4 model
-            auto_translations = [openai.Completion.create(
-                engine="text-davinci-002",  # Change engine if needed
-                prompt=sentence,
-                max_tokens=50  # Adjust max_tokens as needed
-            )['choices'][0]['text'] for sentence in samples_to_translate]
+            # System prompt for translation assistant
+            translation_prompt = """
+            You are a helpful AI-powered translation assistant designed for users seeking reliable translation assistance. Your primary function is to provide context-aware translations from Moroccan Arabic (Darija) to English.
+            """
+
+            auto_translations = []
+
+            for sentence in samples_to_translate:
+                # Create messages for the chat model
+                messages = [
+                    {"role": "system", "content": translation_prompt},
+                    {"role": "user", "content": f"Translate the following sentence to English: '{sentence}'"}
+                ]
+
+                # Perform automatic translation using OpenAI GPT-3.5-turbo model
+                response = openai.ChatCompletion.create(
+                    model="gpt-3.5-turbo",
+                    messages=messages,
+                    api_key=openai_api_key
+                )
+
+                # Extract the translated text from the response
+                translated_text = response.choices[0].message['content'].strip()
+
+                # Append the translated text to the list
+                auto_translations.append(translated_text)
 
             # Update the dataset with auto-translations
             st.session_state.data.loc[