Update app.py

app.py

@@ -22,6 +22,7 @@ loras = [
         "image": "https://cas-bridge.xethub.hf.co/xet-bridge-us/68dc16c261d2d3954790425b/fd0d3426c2bf644e5cb1f57d38c46e1f5e7d73d342ff4de59f32adac598b8c24?X-Amz-Algorithm=AWS4-HMAC-SHA256&X-Amz-Content-Sha256=UNSIGNED-PAYLOAD&X-Amz-Credential=cas%2F20251001%2Fus-east-1%2Fs3%2Faws4_request&X-Amz-Date=20251001T162333Z&X-Amz-Expires=3600&X-Amz-Signature=6e6c04ddf732b5e1d97ac4766f67911fa4437025b4daeaa8c4cb1117dc56e1d6&X-Amz-SignedHeaders=host&X-Xet-Cas-Uid=63114990af1fce227a3f9f66&response-content-disposition=inline%3B+filename*%3DUTF-8%27%271.png%3B+filename%3D%221.png%22%3B&response-content-type=image%2Fpng&x-id=GetObject&Expires=1759339413&Policy=eyJTdGF0ZW1lbnQiOlt7IkNvbmRpdGlvbiI6eyJEYXRlTGVzc1RoYW4iOnsiQVdTOkVwb2NoVGltZSI6MTc1OTMzOTQxM319LCJSZXNvdXJjZSI6Imh0dHBzOi8vY2FzLWJyaWRnZS54ZXRodWIuaGYuY28veGV0LWJyaWRnZS11cy82OGRjMTZjMjYxZDJkMzk1NDc5MDQyNWIvZmQwZDM0MjZjMmJmNjQ0ZTVjYjFmNTdkMzhjNDZlMWY1ZTdkNzNkMzQyZmY0ZGU1OWYzMmFkYWM1OThiOGMyNCoifV19&Signature=aC0bBvnfu55dVC%7E3eAYOBG8dl0uanzBoMTDcPk0zTGYL7EoRDMuVxY3M3%7E1bXE-OFfpxIoftowQr1zwnXm10pPsG-xB4NV8Kx5wBXrPCBZkYU0-vGCXzeXN79kMnAUwQaCU-LyZPM3-TBXnqAvDjXPYAB0nXBo7rkoM8yYv5VhFDm1Ec3dqU0FoFqXYW5Z4vqtiPPCXRpCvM9uRxio7kpwlXO7cg2BVczyYtBjGBBsZyNvcVa8N%7EmzG06TuZxK3bVTQ7TZfrT5XYDP-HYob3O1%7ETlGhzirVprM27tTzrqUmhHnzLCCGPIbGoevWjyPfXiPSw%7EGw%7E3FIHCZZbwS1Nlw__&Key-Pair-Id=K2L8F4GPSG1IFC",
         "trigger_word": "",
         "trigger_position": "",
+        "aspect": "landscape",
         "title": "Peter Chan concept art"
     },
     {
@@ -29,6 +30,7 @@ loras = [
         "image": "https://huggingface.co/lichorosario/lobato-qwen-image-lora/resolve/main/images/milei.png",
         "trigger_word": "",
         "trigger_position": "",
+        "aspect": "landscape",
         "title": "Pablo Lobato"
     },
     {
@@ -36,12 +38,14 @@ loras = [
         "image": "https://huggingface.co/lichorosario/casivaras-qwen-image-lora/resolve/main/images/perros.jpeg",
         "trigger_word": "",
         "trigger_position": "",
+        "aspect": "portrait",
         "title": "Casivaras"
     },
     {
         "repo": "lichorosario/tosti-qwen-image-lora",
         "image": "https://huggingface.co/lichorosario/tosti-qwen-image-lora/resolve/main/images/example_85rpareuq.png",
         "trigger_word": "",
+        "aspect": "square",
         "title": "Tosti"
     },
     {
@@ -49,6 +53,7 @@ loras = [
         "image": "https://huggingface.co/lichorosario/dott-qwen-image-lora/resolve/main/samples/martha.png",
         "trigger_word": "",
         "trigger_position": "",
+        "aspect": "landscape",
         "title": "Day of the Tentacle"
     },
     {
@@ -56,6 +61,7 @@ loras = [
         "image": "https://huggingface.co/lichorosario/samhtr-qwen-image-lora/resolve/main/images/example_irn8g38y0.png",
         "trigger_word": "",
         "trigger_position": "",
+        "aspect": "landscape",
         "title": "Sam and Max hit the road"
     },
     {
@@ -63,6 +69,7 @@ loras = [
         "image": "https://huggingface.co/lichorosario/rtmi-qwen-image-lora/resolve/main/images/lara.png",
         "trigger_word": "",
         "trigger_position": "",
+        "aspect": "landscape",
         "title": "Return to Monkey Island"
     }
 ]
@@ -154,8 +161,10 @@ def update_selection(evt: gr.SelectData, aspect_ratio):
             aspect_ratio = "9:16"
         elif selected_lora["aspect"] == "landscape":
             aspect_ratio = "16:9"
-        else:
+        elif selected_lora["aspect"] == "square":
             aspect_ratio = "1:1"
+        else:
+            aspect_ratio = selected_lora["aspect"]
     
     return (
         gr.update(placeholder=new_placeholder),