22

utils/prompt_utils.py

@@ -12,12 +12,24 @@ def polish_prompt(original_prompt):
         print("Warning: HF_TOKEN is not set. Prompt enhancement is disabled.")
         return original_prompt
 
-    client = InferenceClient(provider="cerebras", api_key=api_key)
-    messages = [{"role": "system", "content": system_prompt}, {"role": "user", "content": original_prompt}]
+    # client = InferenceClient(provider="cerebras", api_key=api_key)
+    # messages = []
+    client = InferenceClient()
+    
+    
     try:
         completion = client.chat.completions.create(
-            model="Qwen/Qwen3-235B-A22B-Instruct-2507", messages=messages
+        rovider="cerebras",
+        api_key=api_key,
+        model="openai/gpt-oss-120b",
+        messages=[
+                {"role": "system", "content": system_prompt}, 
+                {"role": "user", "content": original_prompt}
+            ],
         )
+        # completion = client.chat.completions.create(
+        #     model="Qwen/Qwen3-235B-A22B-Instruct-2507", messages=messages
+        # )
         polished_prompt = completion.choices[0].message.content
         polished_prompt += " Ultra HD, 4K, cinematic composition"
         return polished_prompt.strip().replace("\n", " ")