Update app.py

app.py

@@ -23,7 +23,9 @@ def main():
     st.markdown("*Hint: click on the top-right corner of an image to enlarge it!*")
     # Set the columns
 
-
+    col1, col2 = st.columns(2)
+    col1.subheader("Input page")
+    col2.subheader("Segmentation heatmap")
 
     # Model selection
     st.sidebar.title("Model selection")
@@ -38,16 +40,34 @@ def main():
         with st.spinner("Loading model..."):
             time.sleep(5)
             # load the model TODO
-    
-    st.markdown("# LLM Notebook")
-    with st.form('my_form'):
-        text = st.text_area('Prompt:', placeholder='Please, type your question and submit. ')
-        print("tetx", text)
-        submitted = st.form_submit_button('Submit')
-        if submitted:
+
+    text_input = ''
+    with col1:
+        text_input_temp = st.text_input(
+            "Please, type your question and submit.",
+            "Your prompt",
+            key="placeholder",
+        )
+        
+        if st.button("Submit"):
+            text_input = text_input_temp
+            
+    with col2:
+        if text_input != '':
             with st.spinner("Analyzing..."):
                 time.sleep(4)
-                st.text("You predd submit")
+                st.write("You entered: ", text_input)
+                text_input = ''
+            
+# st.markdown("# LLM Notebook")
+    # with st.form('my_form'):
+    #     text = st.text_area('Prompt:', placeholder='Please, type your question and submit. ')
+    #     print("tetx", text)
+    #     submitted = st.form_submit_button('Submit')
+    #     if submitted:
+    #         with st.spinner("Analyzing..."):
+    #             time.sleep(4)
+    #             st.text("You predd submit")
                 
 
 if __name__ == "__main__":