test gradio

app.py

@@ -82,51 +82,20 @@ gif_pipe.load_lora_weights(lora_adapter_id, adapter_name="motion_lora")
 
 @spaces.GPU
 def generate_gif(prompt, reference_image, controlnet_conditioning_scale,style_conditioning_scale,num_frames):
-    massage_history = [{"role": "system", "content": """
-You are a scene designer tasked with creating sparse frames of a video. You will be given a prompt describing the desired video, and your goal is to design only the key frames (sparse frames) that represent major changes in the scene. Do not include repetitive or similar scenes—only capture distinct moments.
-
-Expected Format:
-Return the response as a JSON object with the key "frames". The value should be a list of dictionaries, where each dictionary has:
-
-"frame_index": an integer indicating the frame's position in the sequence.
-"description": a brief description of the scene in this frame.
-Example:
-If given a prompt like "A sunset over a beach with waves crashing and a ship sailing by," your response should look like this:
-
-```json
-{
-  "frames": [
-    {"frame_index": 0, "description": "Sunset over an empty beach, sky turning orange and pink"},
-    {"frame_index": 30, "description": "Waves gently crashing on the shore"},
-    {"frame_index": 60, "description": "A ship appears on the horizon, silhouetted by the sunset"},
-    {"frame_index": 90, "description": "Ship sailing closer, with waves becoming more dynamic"},
-    {"frame_index": 120, "description": "Sun dipping below the horizon, casting a golden glow over the water"}
-  ]
-}
-```
-This way, each frame represents a distinct scene, and there’s no redundancy between them."""},
-                       {"role": "user", "content": f"give me the frames to generate a video with prompt : `{prompt}`"},]
-    frames = ask_gpt(massage_history,return_str=False)['frames']
-    conditioning_frames = []
-    controlnet_frame_indices =[]
-    frames = frames[0:1]
-    for frame in frames:
-        conditioning_frames.append(generate_image(frame['description'], reference_image, float(style_conditioning_scale)))
-        controlnet_frame_indices.append(frame['frame_index'])
-        yield (conditioning_frames, "output.gif")
+    style_image = generate_image(prompt, reference_image, float(style_conditioning_scale))
 
     video = gif_pipe(
         prompt=prompt,
         negative_prompt="low quality, worst quality",
         num_inference_steps=25,
-        conditioning_frames=conditioning_frames,
-        controlnet_frame_indices=controlnet_frame_indices,
+        conditioning_frames=[style_image],
+        controlnet_frame_indices=[0],
         controlnet_conditioning_scale=float(controlnet_conditioning_scale),
         num_frames=int(num_frames)
     ).frames[0]
     export_to_gif(video, "output.gif")
 
-    yield (conditioning_frames, "output.gif")
+    yield ([style_image], "output.gif")
 
 # Set up Gradio interface
 interface = gr.Interface(