Update app.py

app.py

@@ -49,14 +49,19 @@ def search_reviews(df_original, product_description, without_newlines=False, n=1
 def generate_response(text, model, temperature, top_p):
     reference = search_reviews(data, text, without_newlines=True)["Content"][0]
 
-    completion = client.completions.create(
-        engine=model,
-        prompt=f"Rewrite the following text while ensuring logical composition and avoiding over-exaggeration or imaginative elements: {reference}\n\n{text}",
+    completion = client.chat.completions.create(
+        model=model,
         temperature=temperature,
-        max_tokens=150,  # Вы можете изменить это значение
-        top_p=top_p
+        top_p=top_p,
+        messages=[
+            {
+                "role": "system",
+                "content": f"Generate high-quality rewritten articles, ensuring logical composition, avoiding over-exaggeration, and refraining from any imaginative elements. Use the provided sample text as a reference for the desired writing style:{reference}",
+            },
+            {"role": "user", "content": f"{text}"},
+        ],
     )
-    result = completion.choices[0].text.strip()
+    result = completion.choices[0].message.content
     return result
 
 # Создаём интерфейс Gradio