maybe fix

app.py

@@ -816,86 +816,75 @@ async def generate_piclet(image, hf_token: str) -> dict:
         # Get user profile (creates if doesn't exist)
         user_profile = PicletDiscoveryService.get_or_create_user_profile(user_info)
 
-        # Extract image from Gradio input - save to temp file to pass to API calls
-        # Gradio cleans up temp files, so we save to our own temp file
-        from PIL import Image as PILImage
-        import shutil
-
-        image_path = image if isinstance(image, str) else image.name if hasattr(image, 'name') else str(image)
-
-        # Save to a temporary file that we control
-        with tempfile.NamedTemporaryFile(mode='wb', suffix='.jpg', delete=False) as f:
-            temp_image_path = f.name
-            img = PILImage.open(image_path)
-            img.save(temp_image_path, format='JPEG')
-
-        try:
-            # Step 1: Generate caption
-            print("Step 1/5: Generating image caption...")
-            caption = await PicletGeneratorService.generate_enhanced_caption(temp_image_path, hf_token)
-
-            # Step 2: Generate concept
-            print("Step 2/5: Generating Piclet concept...")
-            concept_data = await PicletGeneratorService.generate_piclet_concept(caption, hf_token)
-
-            object_name = concept_data['objectName']
-            attributes = concept_data['attributes']
-            stats = concept_data['stats']
-            image_prompt = concept_data['imagePrompt']
-            concept_text = concept_data['concept']
-
-            # Step 3: Generate image
-            print("Step 3/5: Generating Piclet image...")
-            image_result = await PicletGeneratorService.generate_piclet_image(
-                image_prompt,
-                stats['tier'],
-                hf_token
-            )
-
-            # Step 4: Check for canonical/variation
-            print("Step 4/5: Checking for existing canonical...")
-            existing_data = PicletDiscoveryService.load_piclet_data(object_name)
-
-            discovery_status = 'new'
-            canonical_id = None
-            scan_count = 1
+        # Extract image path from Gradio input (already a filepath string)
+        image_path = image if isinstance(image, str) else str(image)
+
+        # Step 1: Generate caption
+        print("Step 1/5: Generating image caption...")
+        caption = await PicletGeneratorService.generate_enhanced_caption(image_path, hf_token)
+
+        # Step 2: Generate concept
+        print("Step 2/5: Generating Piclet concept...")
+        concept_data = await PicletGeneratorService.generate_piclet_concept(caption, hf_token)
+
+        object_name = concept_data['objectName']
+        attributes = concept_data['attributes']
+        stats = concept_data['stats']
+        image_prompt = concept_data['imagePrompt']
+        concept_text = concept_data['concept']
+
+        # Step 3: Generate image
+        print("Step 3/5: Generating Piclet image...")
+        image_result = await PicletGeneratorService.generate_piclet_image(
+            image_prompt,
+            stats['tier'],
+            hf_token
+        )
 
-            if existing_data:
-                # Check if this is an exact canonical match (no attributes)
-                if not attributes or len(attributes) == 0:
-                    discovery_status = 'existing'
-                    canonical_id = existing_data['canonical']['typeId']
-                    # Increment scan count
-                    existing_data['canonical']['scanCount'] = existing_data['canonical'].get('scanCount', 0) + 1
-                    scan_count = existing_data['canonical']['scanCount']
-                    PicletDiscoveryService.save_piclet_data(object_name, existing_data)
-                else:
-                    # Check for matching variation
-                    variations = existing_data.get('variations', [])
-                    matched_variation = None
-
-                    for variation in variations:
-                        var_attrs = set(variation.get('attributes', []))
-                        search_attrs = set(attributes)
-                        overlap = len(var_attrs.intersection(search_attrs))
-
-                        if overlap >= len(search_attrs) * 0.5:
-                            matched_variation = variation
-                            discovery_status = 'existing'
-                            canonical_id = existing_data['canonical']['typeId']
-                            # Increment variation scan count
-                            variation['scanCount'] = variation.get('scanCount', 0) + 1
-                            scan_count = variation['scanCount']
-                            PicletDiscoveryService.save_piclet_data(object_name, existing_data)
-                            break
+        # Step 4: Check for canonical/variation
+        print("Step 4/5: Checking for existing canonical...")
+        existing_data = PicletDiscoveryService.load_piclet_data(object_name)
 
-                    if not matched_variation:
-                        discovery_status = 'variation'
+        discovery_status = 'new'
+        canonical_id = None
+        scan_count = 1
+
+        if existing_data:
+            # Check if this is an exact canonical match (no attributes)
+            if not attributes or len(attributes) == 0:
+                discovery_status = 'existing'
+                canonical_id = existing_data['canonical']['typeId']
+                # Increment scan count
+                existing_data['canonical']['scanCount'] = existing_data['canonical'].get('scanCount', 0) + 1
+                scan_count = existing_data['canonical']['scanCount']
+                PicletDiscoveryService.save_piclet_data(object_name, existing_data)
+            else:
+                # Check for matching variation
+                variations = existing_data.get('variations', [])
+                matched_variation = None
+
+                for variation in variations:
+                    var_attrs = set(variation.get('attributes', []))
+                    search_attrs = set(attributes)
+                    overlap = len(var_attrs.intersection(search_attrs))
+
+                    if overlap >= len(search_attrs) * 0.5:
+                        matched_variation = variation
+                        discovery_status = 'existing'
                         canonical_id = existing_data['canonical']['typeId']
+                        # Increment variation scan count
+                        variation['scanCount'] = variation.get('scanCount', 0) + 1
+                        scan_count = variation['scanCount']
+                        PicletDiscoveryService.save_piclet_data(object_name, existing_data)
+                        break
+
+                if not matched_variation:
+                    discovery_status = 'variation'
+                    canonical_id = existing_data['canonical']['typeId']
 
-            # Step 5: Save new discovery if needed
-            print("Step 5/5: Saving to dataset...")
-            if discovery_status == 'new':
+        # Step 5: Save new discovery if needed
+        print("Step 5/5: Saving to dataset...")
+        if discovery_status == 'new':
                 # Create new canonical
                 type_id = f"{PicletDiscoveryService.normalize_object_name(object_name)}_canonical"
                 canonical_data = {