switched to two columns interface

app.py

@@ -29,18 +29,9 @@ def generate_answer(model, processor, image, prompt):
     parsed_answer = processor.post_process_generation(generated_text, task=task, image_size=(image.width, image.height))
     return parsed_answer[task]
 
-# Function to display config without nested expanders
-def display_config(config, depth=0):
-    for key, value in config.items():
-        if isinstance(value, dict):
-            st.markdown(f"{'  ' * depth}**{key}**:")
-            display_config(value, depth + 1)
-        else:
-            st.markdown(f"{'  ' * depth}{key}: {value}")
-
 # Streamlit app
 def main():
-    st.set_page_config(page_title="Lutece-Vision-Base Demo", page_icon="🗼", layout="wide",     initial_sidebar_state="expanded")
+    st.set_page_config(page_title="Lutece-Vision-Base Demo", page_icon="🗼", layout="wide", initial_sidebar_state="expanded")
 
     # Title and description
     st.title("🗼 Lutece-Vision-Base Demo")
@@ -59,20 +50,26 @@ def main():
 
     if uploaded_file is not None:
         image = Image.open(uploaded_file).convert('RGB')
-        st.image(image, caption="Uploaded Document", use_column_width=True)
-
-        # Question input
-        question = st.text_input("❓ Ask a question about the document", "")
+        
+        # Two-column layout
+        col1, col2 = st.columns(2)
+        
+        with col1:
+            # Display image with controlled size
+            st.image(image, caption="Uploaded Document", use_column_width=True)
+        
+        with col2:
+            # Question input
+            question = st.text_input("❓ Ask a question about the document", "")
+            submit_button = st.button("🔍 Generate Answer")
 
-        if st.button("🔍 Generate Answer"):
+        # Answer section spanning both columns
+        if submit_button and question:
             with st.spinner("Generating answer..."):
                 answer = generate_answer(model, processor, image, question)
-            st.success(f"## 💡 {answer}")
-
-    # # Model configuration viewer
-    # with st.expander("🔧 Model Configuration"):
-    #     config_dict = model.config.to_dict()
-    #     display_config(config_dict)
+            st.success("Answer generated!")
+            st.markdown(f"## 💡 Answer")
+            st.markdown(answer)
 
 if __name__ == "__main__":
     main()