changing layout

app.py

@@ -125,15 +125,6 @@ def load_example(example_id):
     return prettify_text(related_chosen)
     # return related_chosen
 
-def generate_similar_summeries():
-    if random.random() > 0.5:
-        return [[ex1], [ex3]]
-    else:
-        return [[ex2], [ex4]]
-
-def remove_info(c1, c2):
-    return gr.ClearButton.update(visible=False)
-
 def prettify_text(nested_list):
     idx = 1
     string = ''
@@ -159,16 +150,16 @@ def main():
         This interface outlines DRGCoder, an explainable clinical coding for the early prediction of diagnostic-related groups (DRGs). Please note all summaries will be truncated to 512 words if longer. 
         """)
         with gr.Row() as row:
-            with gr.Column(scale=2) as col:
-                input = gr.Textbox(label="Input Discharge Summary Here", placeholder='sample discharge summary')
-                btn = gr.Button(value="Submit")
-            with gr.Column(scale=3) as col:
-                gr.Examples(examples, [input])
+            input = gr.Textbox(label="Input Discharge Summary Here", placeholder='sample discharge summary')
         with gr.Row() as row:
-            with gr.Column() as col:
-                attn_viz = gr.HTML() 
-                attn_clr_btn = gr.ClearButton(value='Remove output', visible=False, components=[attn_viz]) 
-                attn_clr_btn.click(clr_btn, outputs=[attn_clr_btn])                    
+            gr.Examples(examples, [input])
+        with gr.Row() as row:
+            btn = gr.Button(value="Submit")
+        with gr.Row() as row:
+            attn_viz = gr.HTML() 
+        with gr.Row() as row:
+            attn_clr_btn = gr.ClearButton(value='Remove output', visible=False, components=[attn_viz]) 
+            attn_clr_btn.click(clr_btn, outputs=[attn_clr_btn])                    
         
         # related row 1
         with gr.Row() as row:
@@ -199,12 +190,13 @@ def main():
 
         # input to related summaries
         with gr.Row() as row:
-            with gr.Column(scale=4) as col:
-                input_related = gr.TextArea(label="Input up to 3 Related Discharge Summary/Summaries Here", visible=False)
-                rmv_related_btn = gr.Button(value='Remove Related Summary', visible=False)
-            with gr.Column(scale=4) as col:
-                related = gr.Dataset(samples=[], components=[input_related], visible=False, type='index')
-                sbm_btn = gr.Button(value="Submit Related Summaries", components=[input_related], visible=False)                              
+            input_related = gr.TextArea(label="Input up to 3 Related Discharge Summary/Summaries Here", visible=False)
+        with gr.Row() as row:
+            rmv_related_btn = gr.Button(value='Remove Related Summary', visible=False)
+            sbm_btn = gr.Button(value="Submit Related Summaries", components=[input_related], visible=False)                              
+        
+        with gr.Row() as row:
+            related = gr.Dataset(samples=[], components=[input_related], visible=False, type='index')
 
         # initial run
         btn.click(run, inputs=[input], outputs=[attn_viz, related, attn_clr_btn, input_related, sbm_btn, rmv_related_btn])