modality order bug fix

app.py

@@ -7,7 +7,7 @@ theme = gr.themes.Soft(primary_hue="cyan", secondary_hue="zinc", font=[gr.themes
 
 with gr.Blocks(theme=theme) as demo:
     with gr.Column(elem_classes="header"):
-        gr.HTML('<img src="https://huggingface.co/spaces/mikonvergence/COP-GEN-Beta/resolve/main/logos/COP-GEN-logo.png" style="max-width: 300px;">')
+        gr.HTML('<img src="https://huggingface.co/spaces/mikonvergence/COP-GEN-Beta/resolve/main/logos/COP-GEN-logo.png" style="max-width: 90%; width:300px;">')
         gr.Markdown("# 🔵 COP-GEN-Beta: Unified Generative Modelling of COPernicus Imagery Thumbnails")
         gr.Markdown("### Miguel Espinosa, Valerio Marsocci, Yuru Jia, Elliot J. Crowley, Mikolaj Czerkawski")
         gr.HTML('<a href="https://github.com/miquel-espinosa/COP-GEN-Beta" class="text-decoration-none site-link"><img alt="GitHub Repo stars" src="https://img.shields.io/github/stars/miquel-espinosa/COP-GEN-Beta"></a>')
@@ -83,4 +83,4 @@ with gr.Blocks(theme=theme) as demo:
         s1rtc_reuse_button.click(fn=pass_value, inputs=[s1rtc_output],outputs=[s1rtc_input])
         dem_reuse_button.click(fn=pass_value, inputs=[dem_output],outputs=[dem_input])
 
-demo.queue().launch()
+demo.queue().launch(share=True)

src/backend.py

@@ -270,11 +270,22 @@ def generate_output(s2l1c_input, s2l2a_input, s1rtc_input, dem_input, num_infere
     output = []
 
     # Collect outputs
-    for modality in sorted_modalities:
-        if modality in input_images:
-            output.append(input_images[modality])
-        else:
-            output.append(to_PIL(imgs_out[modality][0]))
+    if s2l1c_active:
+        output.append(s2l1c_input)
+    else:
+        output.append(to_PIL(imgs_out['s2_l1c'][0]))
+    if s2l2a_active:
+        output.append(s2l2a_input)
+    else:
+        output.append(to_PIL(imgs_out['s2_l2a'][0]))
+    if s1rtc_active:
+        output.append(s1rtc_input)
+    else:
+        output.append(to_PIL(imgs_out['s1_rtc'][0]))
+    if dem_active:
+        output.append(dem_input)
+    else:
+        output.append(to_PIL(imgs_out['dem'][0]))
         
     return output