fix api

app.py

@@ -272,17 +272,18 @@ class PicletGeneratorService:
 
             print(f"Generating caption for image...")
             result = await client.predict(
-                image=image_data,
-                caption_type="Descriptive",
-                caption_length="medium-length",
-                extra_options=[],
-                name_input="",
-                custom_prompt="Describe this image in detail, identifying any recognizable objects, brands, logos, or specific models. Be specific about product names and types.",
-                api_name="/stream_chat"
+                "/stream_chat",
+                image_data,  # input_image
+                "Descriptive",  # caption_type
+                "medium-length",  # caption_length
+                [],  # extra_options
+                "",  # name_input
+                "Describe this image in detail, identifying any recognizable objects, brands, logos, or specific models. Be specific about product names and types."  # custom_prompt
             )
 
-            # JoyCaption returns tuple: (image, caption_text)
-            caption = result[1] if isinstance(result, (list, tuple)) and len(result) > 1 else str(result)
+            # JoyCaption returns tuple: (prompt_used, caption_text) in .data
+            result_data = result.data if hasattr(result, 'data') else result
+            caption = result_data[1] if isinstance(result_data, (list, tuple)) and len(result_data) > 1 else str(result_data)
             print(f"Caption generated: {caption[:100]}...")
             return caption
 
@@ -302,15 +303,16 @@ class PicletGeneratorService:
 
             print(f"Generating text...")
             result = await client.predict(
-                message=prompt,
-                history=[],
-                system_prompt="You are a helpful assistant that creates Pokemon-style monster concepts based on real-world objects.",
-                temperature=0.7,
-                api_name="/chat"
+                "/chat",
+                prompt,  # message
+                [],  # history
+                "You are a helpful assistant that creates Pokemon-style monster concepts based on real-world objects.",  # system_prompt
+                0.7  # temperature
             )
 
             # Extract response text (GPT-OSS formats with Analysis and Response)
-            response_text = result[0] if isinstance(result, (list, tuple)) else str(result)
+            result_data = result.data if hasattr(result, 'data') else result
+            response_text = result_data[0] if isinstance(result_data, (list, tuple)) else str(result_data)
 
             # Try to extract Response section
             response_match = re.search(r'\*\*💬 Response:\*\*\s*\n\n([\s\S]*)', response_text)
@@ -489,18 +491,19 @@ CRITICAL RULES:
 
             print(f"Generating image with prompt: {full_prompt[:100]}...")
             result = await client.predict(
-                prompt=full_prompt,
-                seed=0,
-                randomize_seed=True,
-                width=1024,
-                height=1024,
-                num_inference_steps=4,
-                api_name="/infer"
+                "/infer",
+                full_prompt,  # prompt
+                0,  # seed
+                True,  # randomize_seed
+                1024,  # width
+                1024,  # height
+                4  # num_inference_steps
             )
 
             # Extract image URL and seed
-            image_data = result[0] if isinstance(result, (list, tuple)) else result
-            seed = result[1] if isinstance(result, (list, tuple)) and len(result) > 1 else 0
+            result_data = result.data if hasattr(result, 'data') else result
+            image_data = result_data[0] if isinstance(result_data, (list, tuple)) else result_data
+            seed = result_data[1] if isinstance(result_data, (list, tuple)) and len(result_data) > 1 else 0
 
             # Handle different return formats
             image_url = None