Update app.py

app.py

@@ -10,42 +10,39 @@ def dynamic_relight(image_path, prompt, lighting_choice, s1, s2, s3, s4):
     if image_path is None:
         return None
     
-    # 1. SAFETY: Resize image to prevent remote server timeout
+    # 1. Image preparation
     img = Image.open(image_path)
-    img.thumbnail((512, 512)) # Smaller size ensures higher success rate
-    temp_path = "final_test_input.png"
+    img.thumbnail((512, 512)) # Safety resize for remote processing
+    temp_path = "final_fix.png"
     img.save(temp_path)
 
-    # 2. PROPOSED SYSTEM: User-Desired Keyword Weighting
-    # This creates the 'Unique' project logic you proposed
+    # 2. Unique Project Feature: User-Desired Keyword Weighting
     keywords = ["cinematic", "detailed", "texture", "focus"]
     slider_vals = [s1, s2, s3, s4]
-    
     weighted_prompt = prompt
     for i, word in enumerate(keywords):
         weight = slider_vals[i] / 50.0 
         weighted_prompt += f", ({word}:{weight:.1f})"
     
-    # 3. ROBUST EXECUTION: Positional Call
-    # By analyzing the IC-Light API, the standard input sequence is:
-    # [Image, Prompt, Image Relation, Lighting Preference]
+    # 3. FINAL ACCURATE CALL: Using fn_index=1
+    # This targets the primary 'Relight' function in IC-Light
     try:
-        # We pass arguments as a list to match the remote API schema exactly
-        job = client.submit(
-            handle_file(temp_path), 
-            weighted_prompt,
-            "Appearance Variation", # image_relation
-            lighting_choice,         # lighting_preference
-            api_name="/predict"      # Using the primary verified endpoint
+        result = client.predict(
+            handle_file(temp_path),    # input_fg
+            weighted_prompt,           # prompt
+            "Appearance Variation",     # image_relation
+            lighting_choice,           # lighting_preference
+            fn_index=1                 # Use numerical index to avoid ambiguity
         )
-        return job.result()[0]
+        return result[0]
     except Exception as e:
-        # Fallback to the most basic execution possible
+        # Emergency backup: If the Space updated, try fn_index=2
         return client.predict(
             handle_file(temp_path),
             weighted_prompt,
             "Appearance Variation",
-            lighting_choice
+            lighting_choice,
+            fn_index=2
         )[0]
 
 # UI Setup