Update app.py

app.py

@@ -316,7 +316,7 @@ def get_concat_v_cut(in1, in2, theme='dark'):
 
 
 
-def nasa_sdo_composite(obj,size,date1,date2):
+def nasa_sdo_composite(obj,size,date1,date2,ret_num):
     html_in=f"<style>{css}</style><div class='img_box_soho'>"
     if int(date2[6:8])-int(date1[6:8]) >0:
         return ("<div style='font-size:xxx-large'>Reduce Timespan to 24 hours</div>")
@@ -337,7 +337,7 @@ def nasa_sdo_composite(obj,size,date1,date2):
     anchor_elements = soup.find_all('a')
     #anchor_elements.sort()
     cnt=1
-    max_cnt=4
+    max_cnt=ret_num
     for element in anchor_elements:
         href=element.get('href')
         if href.endswith('.jpg'):
@@ -393,12 +393,12 @@ def get_date(year1,month1,day1,hour1,minute1,year2,month2,day2,hour2,minute2):
     print(date1)
     print(date2)
     return date1,date2
-def comp_combine(inp_ims: list):
+def comp_combine(inp_ims: list,comp_col,comp_row):
     #print(inp_ims)
     im_box=[]
     v_box=[]
-    rows=2
-    cols=2
+    rows=comp_row
+    cols=comp_col
     for i,im in enumerate(eval(inp_ims)):
         urllib.request.urlretrieve(im, f"tmp-{i}.jpg")
         im_box.append(f"tmp-{i}.jpg")
@@ -472,9 +472,9 @@ with gr.Blocks() as app:
             
             with gr.Tab("Composite"):
                 with gr.Row():
-                    comp_num=gr.Number(label="Return",precision=0,value=4)
-                    comp_col=gr.Number(label="Columns",precision=0)
-                    comp_row=gr.Number(label="Rows",precision=0)
+                    comp_num=gr.Number(label="Return",precision=0,value=4,interactive=True)
+                    comp_col=gr.Number(label="Columns",precision=0,value=2,interactive=True)
+                    comp_row=gr.Number(label="Rows",precision=0,value=2,interactive=True)
                 with gr.Row():
                     comp_combine_btn=gr.Button("Combine")
                     sdo_comp_btn=gr.Button("Load")
@@ -498,8 +498,8 @@ with gr.Blocks() as app:
             html_vid_nasa=gr.HTML()
 
     ###### Composite Images ##########
-    comp_combine_btn.click(comp_combine,image_comp_list,img_out)
-    sdo_comp_btn.click(get_date,[year1,month1,day1,hour1,minute1,year2,month2,day2,hour2,minute2],[date1,date2]).then(nasa_sdo_composite,[sdo_drop1,sdo_drop2,date1,date2],[sdo_comp_html,image_comp_list])
+    comp_combine_btn.click(comp_combine,[image_comp_list,comp_col,comp_row],img_out)
+    sdo_comp_btn.click(get_date,[year1,month1,day1,hour1,minute1,year2,month2,day2,hour2,minute2],[date1,date2]).then(nasa_sdo_composite,[sdo_drop1,sdo_drop2,date1,date2,comp_num],[sdo_comp_html,image_comp_list])
 
     ###### NASA Images ##########
     sdo_btn.click(get_date,[year1,month1,day1,hour1,minute1,year2,month2,day2,hour2,minute2],[date1,date2]).then(nasa_sdo_images,[sdo_drop1,sdo_drop2,date1,date2],sdo_html)