Update app.py

app.py

@@ -1,7 +1,9 @@
 import gradio as gr
-from lumaai import LumaAI
+from lumaai.client import LumaAI  # Adjusted based on troubleshooting
 import requests
 import os
+from PIL import Image
+from io import BytesIO
 
 # Initialize LumaAI client
 def create_client(api_key):
@@ -30,6 +32,8 @@ def poll_generation(api_key, generation_id):
         generation = client.generations.get(id=generation_id)
         status = generation.status
         thumbnail_url = generation.assets.thumbnail
+        
+        # Fetch the thumbnail image
         thumbnail_response = requests.get(thumbnail_url)
         thumbnail = thumbnail_response.content if thumbnail_response.status_code == 200 else None
         return status, thumbnail
@@ -139,8 +143,6 @@ with gr.Blocks() as demo:
         
         # Convert thumbnail bytes to image format
         if thumb_data:
-            from PIL import Image
-            from io import BytesIO
             thumb_image = Image.open(BytesIO(thumb_data))
         else:
             thumb_image = None