fix ui issue

chat_column.py

@@ -17,6 +17,7 @@ def render_chat_column(st, llm_client, model_option, max_tokens, BASE_PROMPT):
             st.markdown(message["content"])
 
     # --- Chat Input at the bottom ---
+    # Use a simple approach - just place the input after all messages
     if prompt := st.chat_input("Enter topic to generate JSON for Veo3..."):
         if len(prompt.strip()) == 0:
             st.warning("Please enter a topic.", icon="⚠️")
@@ -84,6 +85,9 @@ def render_chat_column(st, llm_client, model_option, max_tokens, BASE_PROMPT):
                     # Update existing assistant message if needed
                     st.session_state.messages[-1]['content'] = full_response
 
+                # Rerun to ensure proper layout
+                st.rerun()
+
             except Exception as e:
                 st.error(
                     f"Error during LLM response generation: {str(e)}", icon="🚨")