Update content_generation.py

content_generation.py

@@ -113,8 +113,9 @@ def extract_key_contents(script, num_contents=30):
     Trích xuất các ý chính từ script.
     """
     try:
-        response = openai_client.chat.completions.create(
-            model="gpt-3.5-turbo",
+        client = Groq(api_key=GROQ_API_KEY)
+        chat_completion = client.chat.completions.create(
+            model="mixtral-8x7b-32768",
             messages=[
                 {"role": "system", "content": f"Bạn là một chuyên gia phân tích nội dung. Hãy trích xuất chính xác {num_contents} ý chính quan trọng nhất từ đoạn văn sau, mỗi ý không quá 20 từ."},
                 {"role": "user", "content": script}
@@ -122,7 +123,7 @@ def extract_key_contents(script, num_contents=30):
         )
         
         # In response để kiểm tra
-        print("Response từ OpenAI:", response) 
+        print("Response:", response) 
 
         key_contents = response.choices[0].message.content.split('\n')
         return key_contents[:num_contents]
@@ -135,8 +136,9 @@ def generate_image_prompt(content):
     Tạo prompt cho hình ảnh từ nội dung.
     """
     try:
-        response = openai_client.chat.completions.create(
-            model="gpt-3.5-turbo",
+        client = Groq(api_key=GROQ_API_KEY)
+        chat_completion = client.chat.completions.create(
+            model="mixtral-8x7b-32768",
             messages=[
                 {"role": "system", "content": "You are an expert at creating prompts for AI image generation. Create a short, concise prompt in English to visually describe the following content. The content may be in Vietnamese, but your prompt should always be in English."},
                 {"role": "user", "content": content}